This class describes a recurrent (RNN-like) sequence operation. More...
#include <recurrent.hpp>
Public Member Functions | |
RecurrentSequence (OpInputMap &args, ngraph::op::RecurrentSequenceDirection direction) | |
Constructs a RecurrentSequence class object. More... | |
OutputVector | run_sequence (const RecurrentCellFunction &kernel) |
Carry out all steps of recurrent sequence with provided cell kernel. More... | |
This class describes a recurrent (RNN-like) sequence operation.
out on data sequence. Use have to only provide kernel function which would be executed on current time-step input data and the sequence direction mode.
means that the operator should have inputs and outputs the same as RNN operator. Especially the cell/kernel should have input related to hidden cell state.
ngraph::onnx_import::recurrent::RecurrentSequence::RecurrentSequence | ( | OpInputMap & | args, |
ngraph::op::RecurrentSequenceDirection | direction | ||
) |
Constructs a RecurrentSequence class object.
[in] | args | The map with recurrent sequence operator inputs. |
[in] | attrs | The structure containing operator attributes. |
[in] | direction | The sequence direction mode {FORWARD, REVERSE, BIDIRECTIONAL}. |
OutputVector ngraph::onnx_import::recurrent::RecurrentSequence::run_sequence | ( | const RecurrentCellFunction & | kernel | ) |
Carry out all steps of recurrent sequence with provided cell kernel.
[in] | kernel | The cell kernel function. |