ONE - On-device Neural Engine
|
#include <UnidirectionalSequenceLSTM.h>
Public Member Functions | |
LSTMStruct ()=delete | |
LSTMStruct (const LSTMStruct &)=delete | |
LSTMStruct (const circle::Operator *cur_op, luci_interpreter::BaseRuntimeGraph *runtime_graph) | |
void | validateTensorTypes () |
const circle::Tensor * | input () |
const circle::Tensor * | input_to_input_weights () |
const circle::Tensor * | input_to_forget_weights () |
const circle::Tensor * | input_to_cell_weights () |
const circle::Tensor * | input_to_output_weights () |
const circle::Tensor * | recurrent_to_input_weights () |
const circle::Tensor * | recurrent_to_forget_weights () |
const circle::Tensor * | recurrent_to_cell_weights () |
const circle::Tensor * | recurrent_to_output_weights () |
const circle::Tensor * | cell_to_input_weights () |
const circle::Tensor * | cell_to_forget_weights () |
const circle::Tensor * | cell_to_output_weights () |
const circle::Tensor * | input_gate_bias () |
const circle::Tensor * | forget_gate_bias () |
const circle::Tensor * | cell_gate_bias () |
const circle::Tensor * | output_gate_bias () |
const circle::Tensor * | projection_weights () |
const circle::Tensor * | projection_bias () |
const circle::Tensor * | output_state () |
const circle::Tensor * | cell_state () |
const circle::Tensor * | input_layer_norm_coefficients () |
const circle::Tensor * | forget_layer_norm_coefficients () |
const circle::Tensor * | cell_layer_norm_coefficients () |
const circle::Tensor * | output_layer_norm_coefficients () |
const circle::Tensor * | output () |
const circle::Tensor * | get_internal_tensor (int i) |
Data Fields | |
const circle::UnidirectionalSequenceLSTMOptions * | options |
Definition at line 27 of file UnidirectionalSequenceLSTM.h.
|
delete |
|
delete |
|
inlineexplicit |
Definition at line 32 of file UnidirectionalSequenceLSTM.h.
References luci_interpreter::RuntimeGraph::getCircleTensorByIndex(), and options.
|
inline |
Definition at line 163 of file UnidirectionalSequenceLSTM.h.
Referenced by luci_interpreter_pal::lstm_internal::lstmStep().
|
inline |
Definition at line 174 of file UnidirectionalSequenceLSTM.h.
|
inline |
Definition at line 170 of file UnidirectionalSequenceLSTM.h.
|
inline |
Definition at line 158 of file UnidirectionalSequenceLSTM.h.
|
inline |
Definition at line 157 of file UnidirectionalSequenceLSTM.h.
|
inline |
Definition at line 159 of file UnidirectionalSequenceLSTM.h.
|
inline |
Definition at line 162 of file UnidirectionalSequenceLSTM.h.
Referenced by luci_interpreter_pal::lstm_internal::lstmStep(), and validateTensorTypes().
|
inline |
Definition at line 173 of file UnidirectionalSequenceLSTM.h.
|
inline |
Definition at line 180 of file UnidirectionalSequenceLSTM.h.
Referenced by luci_interpreter::configure_kernel_CircleUnidirectionalSequenceLSTM().
|
inline |
Definition at line 145 of file UnidirectionalSequenceLSTM.h.
Referenced by luci_interpreter::configure_kernel_CircleUnidirectionalSequenceLSTM(), luci_interpreter_pal::evalLSTM(), luci_interpreter_pal::evalLSTM< int8_t, int8_t, int16_t, int32_t >(), luci_interpreter_pal::lstm_internal::lstmStep(), and validateTensorTypes().
|
inline |
Definition at line 161 of file UnidirectionalSequenceLSTM.h.
Referenced by luci_interpreter_pal::lstm_internal::lstmStep().
|
inline |
Definition at line 172 of file UnidirectionalSequenceLSTM.h.
|
inline |
Definition at line 149 of file UnidirectionalSequenceLSTM.h.
Referenced by luci_interpreter_pal::lstm_internal::lstmStep().
|
inline |
Definition at line 148 of file UnidirectionalSequenceLSTM.h.
Referenced by luci_interpreter_pal::lstm_internal::lstmStep(), and validateTensorTypes().
|
inline |
Definition at line 147 of file UnidirectionalSequenceLSTM.h.
Referenced by luci_interpreter_pal::lstm_internal::lstmStep().
|
inline |
Definition at line 150 of file UnidirectionalSequenceLSTM.h.
Referenced by luci_interpreter_pal::lstm_internal::lstmStep().
|
inline |
Definition at line 176 of file UnidirectionalSequenceLSTM.h.
Referenced by luci_interpreter_pal::lstm_internal::lstmStep(), and validateTensorTypes().
|
inline |
Definition at line 164 of file UnidirectionalSequenceLSTM.h.
Referenced by luci_interpreter_pal::lstm_internal::lstmStep().
|
inline |
Definition at line 175 of file UnidirectionalSequenceLSTM.h.
|
inline |
Definition at line 169 of file UnidirectionalSequenceLSTM.h.
Referenced by luci_interpreter_pal::evalLSTM(), luci_interpreter_pal::evalLSTM< int8_t, int8_t, int16_t, int32_t >(), and validateTensorTypes().
|
inline |
Definition at line 167 of file UnidirectionalSequenceLSTM.h.
|
inline |
Definition at line 166 of file UnidirectionalSequenceLSTM.h.
|
inline |
Definition at line 154 of file UnidirectionalSequenceLSTM.h.
Referenced by luci_interpreter_pal::lstm_internal::lstmStep().
|
inline |
Definition at line 153 of file UnidirectionalSequenceLSTM.h.
Referenced by luci_interpreter_pal::lstm_internal::lstmStep().
|
inline |
Definition at line 152 of file UnidirectionalSequenceLSTM.h.
Referenced by luci_interpreter_pal::lstm_internal::lstmStep().
|
inline |
Definition at line 155 of file UnidirectionalSequenceLSTM.h.
Referenced by luci_interpreter_pal::lstm_internal::lstmStep().
|
inline |
Definition at line 125 of file UnidirectionalSequenceLSTM.h.
References forget_gate_bias(), input(), input_to_forget_weights(), LUCI_INTERPRETER_CHECK, output(), and output_state().
Referenced by luci_interpreter::configure_kernel_CircleUnidirectionalSequenceLSTM().
const circle::UnidirectionalSequenceLSTMOptions* luci_interpreter::lstm::LSTMStruct::options |
Definition at line 178 of file UnidirectionalSequenceLSTM.h.
Referenced by luci_interpreter::configure_kernel_CircleUnidirectionalSequenceLSTM(), luci_interpreter_pal::evalLSTM(), luci_interpreter_pal::evalLSTM< int8_t, int8_t, int16_t, int32_t >(), and LSTMStruct().