Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Types | Public Member Functions | Protected Attributes
caffe2::Predictor Class Reference

Public Types

using TensorList = std::vector< TensorCPU >
 
using TensorMap = std::unordered_map< std::string, TensorCPU >
 

Public Member Functions

 Predictor (const NetDef &init_net, const NetDef &run_net, Workspace *parent=nullptr, bool run_init=true, int optimization=1)
 
 Predictor (PredictorConfig config)
 
virtual bool operator() (const TensorList &inputs, TensorList *outputs)
 
bool operator() (const TensorMap &inputs, TensorList *outputs)
 
bool operator() (const TensorMap &inputs, TensorMap *outputs)
 
const NetDef & def () const
 
Workspacews ()
 
const std::vector< std::string > & input_names () const
 
const std::vector< std::string > & output_names () const
 

Protected Attributes

PredictorConfig config_
 

Detailed Description

Definition at line 10 of file predictor.h.


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