Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Types | Public Member Functions
caffe2::Predictor Class Reference
Inheritance diagram for caffe2::Predictor:
caffe2::GLPredictor

Public Types

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

Public Member Functions

 Predictor (const MetaNetDef &net, Workspace *parent=nullptr)
 
 Predictor (const NetDef &init_net, const NetDef &run_net, Workspace *parent=nullptr)
 
bool run (const TensorVector &inputs, TensorVector *outputs)
 
bool run_map (const TensorMap &inputs, TensorVector *outputs)
 
const NetDef & def () const
 
Workspacews ()
 

Detailed Description

Definition at line 27 of file predictor.h.


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