|
| NetBase (const std::shared_ptr< const NetDef > &net_def, Workspace *ws) |
|
virtual bool | SupportsAsync ()=0 |
|
const vector< const Event * > & | events () const |
|
virtual void | Wait () |
|
virtual bool | Run () |
|
virtual bool | RunAsync () |
|
virtual vector< float > | TEST_Benchmark (const int, const int, const bool) |
| Benchmarks a network. More...
|
|
const vector< string > & | external_output () const |
|
const vector< string > & | external_input () const |
|
virtual vector< OperatorBase * > | GetOperators () const =0 |
|
const string & | Name () const |
|
const NetDef & | debug_def () const |
|
bool | has_debug_def () const |
|
const Observer * | AttachObserver (std::unique_ptr< Observer > observer) |
|
std::unique_ptr< Observer > | DetachObserver (const Observer *observer_ptr) |
| Returns a unique_ptr to the removed observer. More...
|
|
virtual size_t | NumObservers () |
|
void | StartAllObservers () |
|
void | StopAllObservers () |
|
|
virtual bool | DoRunAsync () |
|
| DISABLE_COPY_AND_ASSIGN (NetBase) |
|
|
vector< string > | external_input_ |
|
vector< string > | external_output_ |
|
string | name_ |
|
vector< const Event * > | events_ |
|
std::shared_ptr< const NetDef > | net_def_ |
|
std::vector< std::unique_ptr< Observer > > | observers_list_ |
|
Definition at line 35 of file net.h.
virtual vector<float> caffe2::NetBase::TEST_Benchmark |
( |
const int |
, |
|
|
const int |
, |
|
|
const bool |
|
|
) |
| |
|
inlinevirtual |
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 in caffe2::DAGNetBase, caffe2::GLNet, caffe2::AsyncSimpleNet, and caffe2::SimpleNet.
Definition at line 77 of file net.h.
The documentation for this class was generated from the following files: