Caffe2 - C++ API
A deep learning, cross platform ML framework
Data Structures | Public Types | Public Member Functions
caffe2::NNApi Class Reference

Public Types

using TensorVector = std::vector< TensorCPU * >
 

Public Member Functions

 NNApi (const NetDef &init_net, const NetDef &run_net, Workspace *ws=nullptr, const PreferenceCode pref=ANEURALNETWORKS_PREFER_SUSTAINED_SPEED)
 
bool loadNNApiLibrary ()
 
bool run (const TensorVector &inputs, TensorVector *outputs)
 

Detailed Description

Definition at line 11 of file nnapi.h.

The documentation for this class was generated from the following files:
Generated on Thu Mar 21 2019 13:06:29 for Caffe2 - C++ API by   doxygen 1.8.11