Searched defs:n_aux_input (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 46 const int n_batch, const int n_input, const int n_aux_input, in CalculateLstmGateFloat()
158 inline void LstmStepCalibration( in LstmStepCalibration()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 49 int n_input, int n_aux_input, int n_output, in ComputeRowSums()
156 const int n_batch, const int n_input, const int n_aux_input, in CalculateLstmGateFloat()
321 const int n_batch, const int n_input, const int n_aux_input, in CalculateLstmGateHybrid()
808 inline void LstmStepFloat( in LstmStepFloat()
977 inline void LstmStepHybrid( in LstmStepHybrid()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 40 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()
|