Searched refs:accumulation_scratch_buffer (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_eval.cc | 889 float* accumulation_scratch_buffer = scratch4; in LstmStepFloat() local
910 accumulation_scratch_buffer, context); in LstmStepFloat()
922 is_aux_input_all_zeros, accumulation_scratch_buffer, context); in LstmStepFloat()
933 is_aux_input_all_zeros, accumulation_scratch_buffer, context); in LstmStepFloat()
948 is_aux_input_all_zeros, accumulation_scratch_buffer, context); in LstmStepFloat()
954 accumulation_scratch_buffer, context); in LstmStepFloat()
1800 float* accumulation_scratch_buffer = nullptr; in EvalFloat() local
1805 accumulation_scratch_buffer = scratch_buffer_ptr + 3 * n_cell * n_batch; in EvalFloat()
1811 accumulation_scratch_buffer = scratch_buffer_ptr + 4 * n_cell * n_batch; in EvalFloat()
1861 output_gate_scratch, accumulation_scratch_buffer, output_ptr, in EvalFloat()
[all …]
|