Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Static Public Member Functions | Protected Attributes
caffe2::Converter Class Referenceabstract

Public Member Functions

void queue (const std::pair< std::string, int > &pair)
void start ()
std::string get ()
void run ()
virtual std::unique_ptr< nom::repr::NeuralNetOperatorconvertToNeuralNetOperator (const OperatorDef &)=0
virtual OperatorDef convertToOperatorDef (const nom::repr::NeuralNetOperator *)

Static Public Member Functions

static std::map< std::string, caffe2::ArgumentgetArgumentsFromOperator (caffe2::OperatorDef op)

Protected Attributes

TensorProtos protos_
TensorProto * data_
TensorProto * label_
std::queue< std::pair< std::string, int > > in_
std::queue< std::string > out_
std::mutex mutex_
std::condition_variable cv_
std::thread thread_

Detailed Description

Definition at line 73 of file

The documentation for this class was generated from the following files: