Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
H A Dlstm.cc40 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()
393 float* input_gate_scratch = nullptr; in EvalFloat() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
H A Dlstm.cc193 float* input_gate_scratch = scratch0; in LstmStepCalibration() local
316 float* input_gate_scratch = nullptr; in EvalCalibration() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dlstm_eval.cc885 float* input_gate_scratch = scratch0; in LstmStepFloat() local
1097 float* input_gate_scratch = scratch0; in LstmStepHybrid() local
1438 int16_t* input_gate_scratch = scratch0; in LstmStepInteger8x8_16() local
1796 float* input_gate_scratch = nullptr; in EvalFloat() local
1998 float* input_gate_scratch = nullptr; in EvalHybrid() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DLSTM.cpp796 float* input_gate_scratch = nullptr; in LSTMStep() local