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

Protected Member Functions

bool RunAt (const std::vector< int > &chain) override
 
- Protected Member Functions inherited from caffe2::DAGNetBase
 DISABLE_COPY_AND_ASSIGN (DAGNetBase)
 
- Protected Member Functions inherited from caffe2::NetBase
 DISABLE_COPY_AND_ASSIGN (NetBase)
 

Additional Inherited Members

- Public Types inherited from caffe2::DAGNetBase
using ExecutionChains = std::unordered_map< int, std::vector< int > >
 
- Public Member Functions inherited from caffe2::DAGNetBase
 DAGNetBase (const NetDef &net_def, Workspace *ws)
 
bool Run () override
 
void WorkerFunction ()
 
vector< float > TEST_Benchmark (const int warmup_runs, const int main_runs, const bool run_individual) override
 Benchmarks a network. More...
 
const ExecutionChains & TEST_execution_chains () const
 
- Public Member Functions inherited from caffe2::NetBase
 NetBase (const NetDef &net_def, Workspace *ws)
 
virtual bool RunAsync ()
 
const vector< string > & external_output () const
 
const vector< string > & external_input () const
 
- Protected Attributes inherited from caffe2::DAGNetBase
vector< internal::OperatorNodeoperator_nodes_
 
ExecutionChains execution_chains_
 
vector< int > initial_frontier_
 
SimpleQueue< int > job_queue_
 
std::vector< std::thread > workers_
 
int num_workers_
 
int remaining_ops_
 
bool success_
 
std::mutex remaining_ops_mutex_
 
std::condition_variable cv_
 
std::mutex run_in_progress_
 
- Protected Attributes inherited from caffe2::NetBase
vector< string > external_input_
 
vector< string > external_output_
 
string name_
 

Detailed Description

Definition at line 569 of file net_dag.cc.


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