Searched refs:LSTMParams (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | LSTM.h | 32 struct LSTMParams { struct
108 const LSTMParams& params, const float* input_buffer, const Shape& input_shape,
133 const LSTMParams& params, const _Float16* input_buffer, const Shape& input_shape,
161 const LSTMParams& params, const float* input_buffer, const Shape& input_shape,
203 uint32_t n_output, uint32_t n_cell, LSTMParams* params);
206 LSTMParams params_;
|
| D | BidirectionalSequenceLSTM.h | 159 LSTMParams params_;
|
| 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()
|
| D | UnidirectionalSequenceLSTM.cpp | 102 inline LSTMParams getLSTMParams(IOperationExecutionContext* context) { in getLSTMParams()
103 LSTMParams params; in getLSTMParams()
|