Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc40 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()
471 float* cell_state_ptr = GetTensorData<float>(cell_state) + b * n_cell; in EvalFloat() local
/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.cc808 inline void LstmStepFloat( in LstmStepFloat()
977 inline void LstmStepHybrid( in LstmStepHybrid()
1316 inline void LstmStepInteger8x8_16( in LstmStepInteger8x8_16()
1569 inline void LstmStepInteger8x8_8( in LstmStepInteger8x8_8()
1823 float* cell_state_ptr = GetTensorData<float>(cell_state) + b * n_cell; in EvalFloat() local
2068 float* cell_state_ptr = GetTensorData<float>(cell_state) + b * n_cell; in EvalHybrid() local
2301 int16_t* cell_state_ptr = in EvalInteger8x8_16() local
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc157 inline void LstmStepCalibration( in LstmStepCalibration()
402 float* cell_state_ptr = GetTensorData<float>(cell_state) + b * n_cell; in EvalCalibration() local