|
using | TensorList = std::vector< TensorCPU > |
|
using | TensorMap = std::unordered_map< std::string, TensorCPU > |
|
|
| 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 |
|
Workspace * | ws () |
|
const std::vector< std::string > & | input_names () const |
|
const std::vector< std::string > & | output_names () const |
|
Definition at line 10 of file predictor.h.
The documentation for this class was generated from the following files: