Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dlstm_eval.cc888 float* output_gate_scratch = scratch3; in LstmStepFloat() local
1100 float* output_gate_scratch = scratch3; in LstmStepHybrid() local
1441 int16_t* output_gate_scratch = scratch3; in LstmStepInteger8x8_16() local
1688 int16_t* output_gate_scratch = scratch4; in LstmStepInteger8x8_8() local
1799 float* output_gate_scratch = nullptr; in EvalFloat() local
2001 float* output_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()
396 float* output_gate_scratch = nullptr; in EvalFloat() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
H A Dlstm.cc196 float* output_gate_scratch = scratch3; in LstmStepCalibration() local
319 float* output_gate_scratch = nullptr; in EvalCalibration() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DLSTM.cpp799 float* output_gate_scratch = nullptr; in LSTMStep() local