Caffe2 - C++ API
A deep learning, cross platform ML framework
Related Pages
Modules
Data Structures
Files
C++ API
Python API
GitHub
File List
Globals
caffe2
mobile
contrib
snpe
snpe_ffi.h
1
#ifndef CAFFE2_SNPE_FFI_H_
2
#define CAFFE2_SNPE_FFI_H_
3
4
#include <stdint.h>
5
#include <string>
6
7
namespace
caffe2
{
8
9
std::string& gSNPELocation();
10
11
const
char
*
const
snpe_ffi_so =
"libsnpe_ffi.so"
;
12
13
}
14
15
extern
"C"
{
16
17
bool
snpe_has_gpu();
18
19
void
* snpe_create(
const
uint8_t* container,
size_t
size,
const
char
* input_name);
20
21
void
snpe_destroy(
void
* ctx);
22
23
void
snpe_get_input_dims(
void
* ctx,
size_t
const
** dims,
size_t
* size);
24
25
void
snpe_run(
void
* ctx,
26
const
float
* inputData,
27
size_t
inputSize,
28
size_t
const
** outputDims,
29
size_t
* outputSize);
30
31
void
snpe_copy_output_to(
void
* ctx,
float
* outputData);
32
33
}
34
35
#endif // CAFFE2_SNPE_FFI_H_
caffe2
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition:
convert_encoded_to_raw_leveldb.cc:47
Generated on Thu Apr 19 2018 13:03:54 for Caffe2 - C++ API by
1.8.11