Searched refs:output_gate_scratch_ptr (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 487 float* output_gate_scratch_ptr = output_gate_scratch + b * n_cell; in EvalFloat() local
517 forget_gate_scratch_ptr, cell_scratch_ptr, output_gate_scratch_ptr, in EvalFloat()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 409 float* output_gate_scratch_ptr = output_gate_scratch + b * n_cell; in EvalCalibration() local
440 output_gate_scratch_ptr, output_ptr, logger, in EvalCalibration()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_eval.cc | 1892 float* output_gate_scratch_ptr = output_gate_scratch + b * n_cell; in EvalFloat() local
1923 output_gate_scratch_ptr, accumulation_scratch_buffer, output_ptr, in EvalFloat()
2138 float* output_gate_scratch_ptr = output_gate_scratch + b * n_cell; in EvalHybrid() local
2194 output_gate_scratch_ptr, GetTensorData<float>(input_sf), in EvalHybrid()
|