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

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: