|
|
| AsyncDAGNet (const std::shared_ptr< const NetDef > &net_def, Workspace *ws) |
| |
|
bool | SupportsAsync () override |
| |
|
bool | RunAt (int chain_id, const std::vector< int > &chain) override |
| |
|
| DAGNetBase (const std::shared_ptr< 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 dag_utils::ExecutionChains & | TEST_execution_chains () const |
| |
|
vector< OperatorBase * > | GetOperators () const override |
| |
|
| NetBase (const std::shared_ptr< const NetDef > &net_def, Workspace *ws) |
| |
|
const vector< const Event * > & | events () const |
| |
|
virtual void | Wait () |
| |
|
virtual bool | Run () |
| |
|
virtual bool | RunAsync () |
| |
|
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 |
| |
|
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 () |
| |
|
|
bool | DoRunAsync () override |
| |
|
int | stream (const DeviceOption &device_option) |
| |
|
| DISABLE_COPY_AND_ASSIGN (AsyncDAGNet) |
| |
|
bool | DoRunAsync () override |
| |
|
void | HandleException (int operator_idx, const std::string &exception_str) |
| |
|
| DISABLE_COPY_AND_ASSIGN (DAGNetBase) |
| |
|
| DISABLE_COPY_AND_ASSIGN (NetBase) |
| |
|
|
std::vector< int32_t > | eventRecorded_ |
| |
|
vector< dag_utils::OperatorNode > | operator_nodes_ |
| |
|
vector< OperatorBase * > | operators_ |
| |
|
dag_utils::ExecutionChains | execution_chains_ |
| |
|
vector< int > | initial_frontier_ |
| |
|
std::unique_ptr< SimpleQueue< int > > | job_queue_ |
| |
|
std::vector< std::thread > | workers_ |
| |
|
int | num_workers_ |
| |
|
int | remaining_ops_ |
| |
|
bool | success_ |
| |
|
std::atomic< bool > | caught_exception_yet_ |
| |
|
int | iter_ |
| |
|
std::mutex | remaining_ops_mutex_ |
| |
|
std::condition_variable | cv_ |
| |
|
std::mutex | run_in_progress_ |
| |
|
std::vector< DAGNetStats > | stats_ |
| |
|
std::unordered_map< int, std::unique_ptr< Timer > > | task_timers_ |
| |
|
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_ |
| |
|
|
static thread_local std::vector< int > | stream_counters_ |
| |
Definition at line 17 of file net_async_dag_gpu.h.
The documentation for this class was generated from the following files:
Public Member Functions inherited from caffe2::DAGNetBase
1.8.11
