Inheritance diagram for caffe2::ParallelNet:

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

TaskThreadPoolBase *	Pool (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

Observable &	operator= (Observable &&)=default

	C10_DISABLE_COPY_AND_ASSIGN (Observable)

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 ()

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 ()

TaskThreadPoolBase *	poolGetter (PoolsMap &pools, int device_type, int device_id, int pool_size)

	C10_DISABLE_COPY_AND_ASSIGN (ParallelNet)

Protected Member Functions inherited from caffe2::NetBase
virtual bool	DoRunAsync ()

	C10_DISABLE_COPY_AND_ASSIGN (NetBase)

Protected Attributes
ExecutionOptions	options_

int	num_workers_

std::unique_ptr< ParallelNetExecutorHelper >	helper_

std::shared_ptr< AsyncTaskGraphBase >	task_graph_

AsyncTaskFuture *	run_future_

std::vector< dag_utils::OperatorNode >	operator_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_

Friends
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:

Public Member Functions