Searched defs:n_aux_input (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_eval.cc | 80 int n_input, int n_aux_input, int n_output, in ComputeRowSums()
187 const int n_batch, const int n_input, const int n_aux_input, in CalculateLstmGateFloat()
363 const int n_batch, const int n_input, const int n_aux_input, in CalculateLstmGateHybrid()
851 inline void LstmStepFloat( in LstmStepFloat()
1033 inline void LstmStepHybrid( in LstmStepHybrid()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceLSTM.cpp | 341 uint32_t n_aux_input = SizeOfDimension(input_, 2); in Prepare() local
|
| D | LSTM.cpp | 793 const uint32_t n_aux_input = aux_input_buffer == nullptr ? 0 : n_input; in LSTMStep() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 40 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()
|