Inheritance diagram for caffe2::AsyncDAGNet:

Public Member Functions
	AsyncDAGNet (const NetDef &net_def, Workspace *ws)

bool	RunAt (const std::vector< int > &chain) override

bool	Run () override

Public Member Functions inherited from caffe2::DAGNetBase
	DAGNetBase (const NetDef &net_def, Workspace *ws)

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 Member Functions
	DISABLE_COPY_AND_ASSIGN (AsyncDAGNet)

Protected Member Functions inherited from caffe2::DAGNetBase
	DISABLE_COPY_AND_ASSIGN (DAGNetBase)

Protected Member Functions inherited from caffe2::NetBase
	DISABLE_COPY_AND_ASSIGN (NetBase)

Protected Attributes
std::vector< int32_t >	eventRecorded_

std::vector< std::unique_ptr< internal::Event > >	events_

Protected Attributes inherited from caffe2::DAGNetBase
vector< internal::OperatorNode >	operator_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_

Additional Inherited Members
Public Types inherited from caffe2::DAGNetBase
using	ExecutionChains = std::unordered_map< int, std::vector< int > >

Detailed Description

Definition at line 36 of file net_gpu.h.

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

Public Member Functions