Searched refs:scratch_buffer_buffer (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | LSTM.h | 129 float* cell_state_out_buffer, float* output_buffer, float* scratch_buffer_buffer,
158 _Float16* scratch_buffer_buffer, bool timeMajor = true, bool forwardSequence = true);
182 float* cell_state_out_buffer, float* output_buffer, float* scratch_buffer_buffer);
|
| D | LSTM.cpp | 425 float* cell_state_out_buffer, float* output_buffer, float* scratch_buffer_buffer, in LSTMEvalFloat32() argument
500 scratch_buffer_buffer); in LSTMEvalFloat32()
545 _Float16* cell_state_out_buffer, _Float16* output_buffer, _Float16* scratch_buffer_buffer, in LSTMEvalFloat16()
678 convertFloat16ToFloat32(scratch_buffer_buffer, &scratch_buffer_float32); in LSTMEvalFloat16()
760 convertFloat32ToFloat16(scratch_buffer_float32, scratch_buffer_buffer); in LSTMEvalFloat16()
785 float* cell_state_out_buffer, float* output_buffer, float* scratch_buffer_buffer) { in LSTMStep() argument
801 cell_scratch = scratch_buffer_buffer; in LSTMStep()
805 input_gate_scratch = scratch_buffer_buffer; in LSTMStep()
|