caffe2::Predictor Class Reference

Inheritance diagram for caffe2::Predictor:

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
Workspace *	ws ()

Detailed Description

Definition at line 11 of file predictor.h.

---

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

• caffe2/core/predictor.h
• caffe2/core/predictor.cc