|
| 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...
|
|
| NetBase (const NetDef &net_def, Workspace *ws) |
|
const vector< string > & | external_output () const |
|
const vector< string > & | external_input () const |
|
|
vector< unique_ptr< OperatorBase > > | operators_ |
|
vector< string > | external_input_ |
|
vector< string > | external_output_ |
|
string | name_ |
|
Definition at line 91 of file net.h.
◆ 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: