Home
last modified time | relevance | path

Searched defs:forget_gate_scratch (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dlstm_eval.cc886 float* forget_gate_scratch = scratch1; in LstmStepFloat() local
1098 float* forget_gate_scratch = scratch1; in LstmStepHybrid() local
1439 int16_t* forget_gate_scratch = scratch1; in LstmStepInteger8x8_16() local
1686 int16_t* forget_gate_scratch = scratch2; in LstmStepInteger8x8_8() local
1798 float* forget_gate_scratch = nullptr; in EvalFloat() local
2000 float* forget_gate_scratch = nullptr; in EvalHybrid() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
H A Dlstm.cc40 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()
395 float* forget_gate_scratch = nullptr; in EvalFloat() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
H A Dlstm.cc194 float* forget_gate_scratch = scratch1; in LstmStepCalibration() local
318 float* forget_gate_scratch = nullptr; in EvalCalibration() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DLSTM.cpp798 float* forget_gate_scratch = nullptr; in LSTMStep() local