Home
last modified time | relevance | path

Searched refs:forget_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 Dlstm.cc485 float* forget_gate_scratch_ptr = forget_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 Dlstm.cc407 float* forget_gate_scratch_ptr = forget_gate_scratch + b * n_cell; in EvalCalibration() local
439 forget_gate_scratch_ptr, cell_gate_scratch_ptr, in EvalCalibration()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dlstm_eval.cc1890 float* forget_gate_scratch_ptr = forget_gate_scratch + b * n_cell; in EvalFloat() local
1922 forget_gate_scratch_ptr, cell_gate_scratch_ptr, in EvalFloat()
2136 float* forget_gate_scratch_ptr = forget_gate_scratch + b * n_cell; in EvalHybrid() local
2193 forget_gate_scratch_ptr, cell_gate_scratch_ptr, in EvalHybrid()