Struct caffe2op_rnn::ThreadedRecurrentNetworkExecutor

pub struct ThreadedRecurrentNetworkExecutor { /* private fields */ }

---

Implementations

impl ThreadedRecurrentNetworkExecutor

pub fn new( step_net_def: &NetDef, recurrent_input_map: &mut HashMap<String, String>, timestep_blob: String ) -> Self

pub fn ignore_link_dependencies(&mut self) -> bool

pub fn set_num_threads(&mut self, n: i32)

pub fn exec_range(&mut self, from: i32, to: i32)

pub fn run(&mut self, t: i32) -> bool

| Run forwardpass with T timesteps. |

pub fn run_backwards(&mut self, t: i32) -> bool

| Run backward pass with T timesteps. |

pub fn run_op(&mut self, job: OpTask, thread_id: i32)

| Runs a single op and updates its dependencies | when finished. | | If dependent ops are ready to run, adds | them to the task_queue. |

pub fn worker_function(&mut self)

| Run-loop for executor threads: pop | tasks from task_queue and execute them | with RunOp(). |

pub fn exec(&mut self)

| Start worker threads if not started | yet, wait until all tasks finished, | or a failure. Called by Run() and RunBackwards(). |

Trait Implementations

impl Drop for ThreadedRecurrentNetworkExecutor

fn drop(&mut self)

Executes the destructor for this type.