Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
caffe2::ThreadedRecurrentNetworkExecutor Class Reference
Inheritance diagram for caffe2::ThreadedRecurrentNetworkExecutor:
caffe2::RecurrentNetworkExecutorBase

Public Member Functions

 ThreadedRecurrentNetworkExecutor (const NetDef &step_net_def, std::map< string, string > &recurrent_input_map, std::string timestep_blob)
 
bool Run (int T) override
 Run forwardpass with T timesteps.
 
bool RunBackwards (int T) override
 Run backward pass with T timesteps.
 
bool ignoreLinkDependencies () override
 
void setNumThreads (int n)
 
- Public Member Functions inherited from caffe2::RecurrentNetworkExecutorBase
void EnsureTimestepInitialized (int t, Workspace *ws, const std::vector< std::unique_ptr< ObserverBase< OperatorBase >>> &observers_list)
 Callers must call EnsureTimestepInitialized before starting execution for each of the relevant timesteps. More...
 
void SetMaxParallelTimesteps (int p)
 Set limit for the number of timesteps that run in parallel. More...
 
size_t NumObserversStepNet ()
 

Additional Inherited Members

- Data Fields inherited from caffe2::RecurrentNetworkExecutorBase
bool debug_ = false
 
- Protected Member Functions inherited from caffe2::RecurrentNetworkExecutorBase
 RecurrentNetworkExecutorBase (const NetDef &step_net_def, std::map< string, string > &recurrent_input_map, std::string timestep_blob)
 
void PrintInfo (int t)
 For debug purposes, print the dependency structure. More...
 
virtual void AnalyzeOps ()
 
- Protected Attributes inherited from caffe2::RecurrentNetworkExecutorBase
std::vector< std::vector< RNNNetOperator > > timestep_ops_
 
std::vector< OperatorBase * > op_ptrs_
 
std::vector< RNNNetOperatortimestep_ops_template_
 
NetDef step_net_def_
 
std::vector< std::vector< string > > op_deps_
 
std::vector< Workspace * > workspaces_
 
std::map< string, string > recurrent_input_map_
 
std::string timestep_blob_
 
int max_parallel_timesteps_ = -1
 

Detailed Description

Definition at line 478 of file recurrent_network_executor.h.


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