|
| 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) |
|
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 () |
|
|
bool | debug_ = false |
|
| 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 () |
|
std::vector< std::vector< RNNNetOperator > > | timestep_ops_ |
|
std::vector< OperatorBase * > | op_ptrs_ |
|
std::vector< RNNNetOperator > | timestep_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 |
|
The documentation for this class was generated from the following files: