Searched refs:LSTMParams (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | LSTM.h | 31 struct LSTMParams { struct
107 const LSTMParams& params, const float* input_buffer, const Shape& input_shape,
132 const LSTMParams& params, const _Float16* input_buffer, const Shape& input_shape,
160 const LSTMParams& params, const float* input_buffer, const Shape& input_shape,
202 uint32_t n_output, uint32_t n_cell, LSTMParams* params);
205 LSTMParams params_;
|
| D | BidirectionalSequenceLSTM.h | 156 LSTMParams params_;
|
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 49 inline LSTMParams getLSTMParams(IOperationExecutionContext* context) { in getLSTMParams()
50 LSTMParams params; in getLSTMParams()
|
| D | LSTM.cpp | 151 uint32_t n_cell, LSTMParams* params) { in CheckInputTensorDimensions()
406 const LSTMParams& params, const float* input_buffer, const Shape& input_shape, in LSTMEvalFloat32()
522 const LSTMParams& params, const _Float16* input_buffer, const Shape& input_shape, in LSTMEvalFloat16()
766 const LSTMParams& params, const float* input_buffer, const Shape& input_shape, in LSTMStep()
|