Searched refs:input_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 | 483 float* input_gate_scratch_ptr = in EvalFloat() local
516 activation_state_ptr, cell_state_ptr, input_gate_scratch_ptr, in EvalFloat()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 405 float* input_gate_scratch_ptr = in EvalCalibration() local
438 output_state_ptr, cell_state_ptr, input_gate_scratch_ptr, in EvalCalibration()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_eval.cc | 1888 float* input_gate_scratch_ptr = in EvalFloat() local
1921 output_state_ptr, cell_state_ptr, input_gate_scratch_ptr, in EvalFloat()
2134 float* input_gate_scratch_ptr = in EvalHybrid() local
2192 output_batch_leading_dim, input_gate_scratch_ptr, in EvalHybrid()
|