Searched refs:input_buffer (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | LSTM.h | 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,
|
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LSTM.cpp | 406 const LSTMParams& params, const float* input_buffer, const Shape& input_shape, in LSTMEvalFloat32() argument
453 transposeFirstTwoDimensions<float>(input_buffer, input_shape, transposedInput.data()); in LSTMEvalFloat32()
464 const float* inputData = timeMajor ? input_buffer : transposedInput.data(); in LSTMEvalFloat32()
522 const LSTMParams& params, const _Float16* input_buffer, const Shape& input_shape, in LSTMEvalFloat16() argument
566 convertFloat16ToFloat32(input_buffer, &input_float32); in LSTMEvalFloat16()
766 const LSTMParams& params, const float* input_buffer, const Shape& input_shape, in LSTMStep() argument
836 n_cell, n_input, input_buffer, in LSTMStep()
840 n_cell, n_input, input_buffer, in LSTMStep()
843 input_to_cell_weights_buffer, n_cell, n_input, input_buffer, n_batch, cell_scratch); in LSTMStep()
845 n_cell, n_input, input_buffer, in LSTMStep()
|