Home
last modified time | relevance | path

Searched defs:cell_state_ptr (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dlstm_eval.cc851 inline void LstmStepFloat( in LstmStepFloat()
1033 inline void LstmStepHybrid( in LstmStepHybrid()
1372 inline void LstmStepInteger8x8_16( in LstmStepInteger8x8_16()
1627 inline void LstmStepInteger8x8_8( in LstmStepInteger8x8_8()
1886 float* cell_state_ptr = GetTensorData<float>(cell_state) + b * n_cell; in EvalFloat() local
2132 float* cell_state_ptr = GetTensorData<float>(cell_state) + b * n_cell; in EvalHybrid() local
2365 int16_t* cell_state_ptr = in EvalInteger8x8_16() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
H A Dlstm.cc40 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()
481 float* cell_state_ptr = GetTensorData<float>(cell_state) + b * n_cell; in EvalFloat() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
H A Dlstm.cc158 inline void LstmStepCalibration( in LstmStepCalibration()
403 float* cell_state_ptr = GetTensorData<float>(cell_state) + b * n_cell; in EvalCalibration() local