ONE - On-device Neural Engine
|
#include <PALUnidirectionalSequenceLSTMCommon.h>
Data Fields | |
bool | time_major |
int32_t | batch_size |
int32_t | time_steps |
int32_t | input_dimension |
int32_t | state_dimension |
Definition at line 181 of file PALUnidirectionalSequenceLSTMCommon.h.
int32_t luci_interpreter_pal::lstm_internal::LstmSizeInfo::batch_size |
Definition at line 184 of file PALUnidirectionalSequenceLSTMCommon.h.
Referenced by luci_interpreter_pal::evalLSTM(), luci_interpreter_pal::evalLSTM< int8_t, int8_t, int16_t, int32_t >(), luci_interpreter_pal::lstm_internal::LstmStepManager::inputShape(), luci_interpreter_pal::lstm_internal::LstmStepManager::stateShape(), and luci_interpreter_pal::lstm_internal::LstmStepManager::updateTime().
int32_t luci_interpreter_pal::lstm_internal::LstmSizeInfo::input_dimension |
Definition at line 186 of file PALUnidirectionalSequenceLSTMCommon.h.
Referenced by luci_interpreter_pal::evalLSTM(), luci_interpreter_pal::evalLSTM< int8_t, int8_t, int16_t, int32_t >(), luci_interpreter_pal::lstm_internal::LstmStepManager::inputShape(), and luci_interpreter_pal::lstm_internal::LstmStepManager::updateTime().
int32_t luci_interpreter_pal::lstm_internal::LstmSizeInfo::state_dimension |
Definition at line 187 of file PALUnidirectionalSequenceLSTMCommon.h.
Referenced by luci_interpreter_pal::evalLSTM(), luci_interpreter_pal::evalLSTM< int8_t, int8_t, int16_t, int32_t >(), luci_interpreter_pal::lstm_internal::LstmStepManager::stateShape(), luci_interpreter_pal::lstm_internal::LstmStepManager::updateBatch(), and luci_interpreter_pal::lstm_internal::LstmStepManager::updateTime().
bool luci_interpreter_pal::lstm_internal::LstmSizeInfo::time_major |
Definition at line 183 of file PALUnidirectionalSequenceLSTMCommon.h.
Referenced by luci_interpreter_pal::evalLSTM(), luci_interpreter_pal::evalLSTM< int8_t, int8_t, int16_t, int32_t >(), luci_interpreter_pal::lstm_internal::LstmStepManager::inputShape(), luci_interpreter_pal::lstm_internal::LstmStepManager::stateShape(), luci_interpreter_pal::lstm_internal::LstmStepManager::updateBatch(), and luci_interpreter_pal::lstm_internal::LstmStepManager::updateTime().
int32_t luci_interpreter_pal::lstm_internal::LstmSizeInfo::time_steps |
Definition at line 185 of file PALUnidirectionalSequenceLSTMCommon.h.
Referenced by luci_interpreter_pal::evalLSTM(), and luci_interpreter_pal::evalLSTM< int8_t, int8_t, int16_t, int32_t >().