Searched refs:output_buffer (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | LSTM.cpp | 382 float* cell_state_out_buffer, float* output_buffer, float* scratch_buffer_buffer, in LSTMEvalFloat32() argument
424 float* outputData = timeMajor ? output_buffer : transposedOutput.data(); in LSTMEvalFloat32()
471 output_buffer); in LSTMEvalFloat32()
502 _Float16* cell_state_out_buffer, _Float16* output_buffer, _Float16* scratch_buffer_buffer, in LSTMEvalFloat16()
632 convertFloat16ToFloat32(output_buffer, &output_float32); in LSTMEvalFloat16()
716 convertFloat32ToFloat16(output_float32, output_buffer); in LSTMEvalFloat16()
742 float* cell_state_out_buffer, float* output_buffer, float* scratch_buffer_buffer) { in LSTMStep() argument
933 output_buffer); in LSTMStep()
935 tflite::tensor_utils::ZeroVector(output_buffer, n_batch * n_output); in LSTMStep()
939 output_buffer, in LSTMStep()
[all …]
|
| D | LSTM.h | 129 float* cell_state_out_buffer, float* output_buffer, float* scratch_buffer_buffer,
157 _Float16* cell_state_out_buffer, _Float16* output_buffer,
182 float* cell_state_out_buffer, float* output_buffer, float* scratch_buffer_buffer);
|