Searched refs:accum_scratch_ptr (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | bidirectional_sequence_rnn.cc | 578 int32_t* accum_scratch_ptr = GetTensorData<int32_t>(accum_scratch); in EvalHybrid() local
616 accum_scratch_ptr, fw_row_sums_ptr, fw_compute_row_sums); in EvalHybrid()
643 accum_scratch_ptr, bw_row_sums_ptr, bw_compute_row_sums); in EvalHybrid()
674 accum_scratch_ptr, fw_row_sums_ptr, fw_compute_row_sums); in EvalHybrid()
705 accum_scratch_ptr, bw_row_sums_ptr, bw_compute_row_sums); in EvalHybrid()
|
| H A D | unidirectional_sequence_rnn.cc | 300 int32_t* accum_scratch_ptr = GetTensorData<int32_t>(accum_scratch); in EvalHybrid() local
327 accum_scratch_ptr, row_sums_ptr, compute_row_sums); in EvalHybrid()
349 accum_scratch_ptr, row_sums_ptr, compute_row_sums); in EvalHybrid()
|
| H A D | basic_rnn.cc | 263 int32_t* accum_scratch_ptr = GetTensorData<int32_t>(accum_scratch); in EvalHybrid() local
276 params->asymmetric_quantize_inputs, zero_points_ptr, accum_scratch_ptr, in EvalHybrid()
|
| H A D | lstm_eval.cc | 1087 float* output_state_ptr, float* cell_state_ptr, int32_t* accum_scratch_ptr, in LstmStepHybrid()
1209 scaling_factors_scratch, recovered_cell_weights, accum_scratch_ptr); in LstmStepHybrid()
1226 scaling_factors_scratch, recovered_cell_weights, accum_scratch_ptr); in LstmStepHybrid()
1243 accum_scratch_ptr); in LstmStepHybrid()
1263 scaling_factors_scratch, recovered_cell_weights, accum_scratch_ptr); in LstmStepHybrid()
1271 input_zp, accum_scratch_ptr); in LstmStepHybrid()
|