Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Protected Member Functions | Protected Attributes
caffe2::SimpleNet Class Reference
Inheritance diagram for caffe2::SimpleNet:
caffe2::NetBase

Public Member Functions

 SimpleNet (const NetDef &net_def, Workspace *ws)
 
bool Run () override
 
bool RunAsync () override
 
vector< float > TEST_Benchmark (const int warmup_runs, const int main_runs, const bool run_individual) override
 Benchmarks a network. More...
 
- Public Member Functions inherited from caffe2::NetBase
 NetBase (const NetDef &net_def, Workspace *ws)
 
const vector< string > & external_output () const
 
const vector< string > & external_input () const
 

Protected Member Functions

 DISABLE_COPY_AND_ASSIGN (SimpleNet)
 
- Protected Member Functions inherited from caffe2::NetBase
 DISABLE_COPY_AND_ASSIGN (NetBase)
 

Protected Attributes

vector< unique_ptr< OperatorBase > > operators_
 
- Protected Attributes inherited from caffe2::NetBase
vector< string > external_input_
 
vector< string > external_output_
 
string name_
 

Detailed Description

Definition at line 91 of file net.h.

Member Function Documentation

◆ TEST_Benchmark()

vector< float > caffe2::SimpleNet::TEST_Benchmark ( const int  warmup_runs,
const int  main_runs,
const bool  run_individual 
)
overridevirtual

Benchmarks a network.

This function returns a vector of float recording the number of milli- seconds spent during the benchmark. The 0-th item is the time spent per each network run, and if a net instantiation supports run_individual, the remainder of the vector returns the number of milliseconds spent per opeartor.

Reimplemented from caffe2::NetBase.

Definition at line 131 of file net.cc.


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