Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Protected Types | Protected Member Functions | Protected Attributes | Friends
caffe2::ParallelNet Class Reference
Inheritance diagram for caffe2::ParallelNet:
caffe2::NetBase caffe2::Observable< NetBase >

Public Member Functions

 ParallelNet (const std::shared_ptr< const NetDef > &net_def, Workspace *ws)
bool RunAsync () override
void Wait () override
bool SupportsAsync () override
std::vector< OperatorBase * > GetOperators () const override
TaskThreadPoolBasePool (const DeviceOption &device_option)
- Public Member Functions inherited from caffe2::NetBase
 NetBase (const std::shared_ptr< const NetDef > &net_def, Workspace *ws)
const vector< const Event * > & events () const
virtual bool Run ()
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
const string & Name () const
const NetDef & debug_def () const
bool has_debug_def () const
- Public Member Functions inherited from caffe2::Observable< NetBase >
 Observable (Observable &&)=default
Observableoperator= (Observable &&)=default
const ObserverAttachObserver (std::unique_ptr< Observer > observer)
std::unique_ptr< ObserverDetachObserver (const Observer *observer_ptr)
 Returns a unique_ptr to the removed observer. More...
virtual size_t NumObservers ()
void StartAllObservers ()
void StopAllObservers ()

Protected Types

typedef std::unordered_map< int, std::unordered_map< int, std::shared_ptr< TaskThreadPoolBase > > > PoolsMap

Protected Member Functions

bool handleRunError () override
virtual void finishRun ()
virtual void reset ()
TaskThreadPoolBasepoolGetter (PoolsMap &pools, int device_type, int device_id, int pool_size)
- Protected Member Functions inherited from caffe2::NetBase
virtual bool DoRunAsync ()

Protected Attributes

ExecutionOptions options_
int num_workers_
std::unique_ptr< ParallelNetExecutorHelperhelper_
std::shared_ptr< AsyncTaskGraphBasetask_graph_
std::vector< dag_utils::OperatorNodeoperator_nodes_
std::vector< OperatorBase * > operators_
std::mutex pools_mutex_
PoolsMap cpu_pools_
PoolsMap gpu_pools_
- Protected Attributes inherited from caffe2::NetBase
vector< string > external_input_
vector< string > external_output_
string name_
vector< const Event * > events_
std::shared_ptr< const NetDef > net_def_
- Protected Attributes inherited from caffe2::Observable< NetBase >
std::vector< std::unique_ptr< Observer > > observers_list_


class ParallelNetExecutorHelper

Additional Inherited Members

- Public Types inherited from caffe2::Observable< NetBase >
using Observer = ObserverBase< NetBase >

Detailed Description

Definition at line 13 of file net_parallel.h.

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