1 #ifndef NNAPI_MODEL_LOADER_H_ 2 #define NNAPI_MODEL_LOADER_H_ 3 4 #include <stdint.h> 5 6 #include <ATen/nnapi/NeuralNetworks.h> 7 #include <ATen/nnapi/nnapi_wrapper.h> 8 9 namespace caffe2 { 10 namespace nnapi { 11 12 int load_nnapi_model( 13 struct nnapi_wrapper* nnapi, 14 ANeuralNetworksModel* model, 15 const void* serialized_model, 16 int64_t model_length, 17 size_t num_buffers, 18 const void** buffer_ptrs, 19 int32_t* buffer_sizes, 20 size_t num_memories, 21 ANeuralNetworksMemory** memories, 22 int32_t* memory_sizes, 23 int32_t* out_input_count, 24 int32_t* out_output_count, 25 size_t* out_bytes_consumed); 26 27 }} // namespace caffe2::nnapi 28 29 #endif // NNAPI_MODEL_LOADER_H_ 30