Caffe2 - C++ API
A deep learning, cross platform ML framework
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_
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition: blob.h:13