Searched defs:aux_input_size (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | kernel_utils.cc | 42 int input_size, int aux_input_size, int num_units, in RnnBatchStep()
148 const float* bias_ptr, int input_size, int aux_input_size, int num_units, in RnnBatchStep()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | bidirectional_sequence_rnn_test.cc | 666 int bw_units, int input_size, int aux_input_size, in BidirectionalRNNOpModel()
817 int aux_input_size() { return aux_input_size_; } in aux_input_size() function in tflite::__anona5c2041c0111::BidirectionalRNNOpModel
1421 void run_closedbox_test_with_input_split(int input_size, int aux_input_size) { in run_closedbox_test_with_input_split()
|
| H A D | bidirectional_sequence_rnn.cc | 393 const int aux_input_size = (aux_input) ? aux_input->dims->data[2] : 0; in EvalFloat() local
539 const int aux_input_size = (aux_input) ? aux_input->dims->data[2] : 0; in EvalHybrid() local
|
| H A D | lstm_eval.cc | 1783 const int aux_input_size = in EvalFloat() local
1987 const int aux_input_size = in EvalHybrid() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | RNN.cpp | 151 uint32_t aux_input_size = 0; in RNNStep() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 380 const int aux_input_size = in EvalFloat() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 303 const int aux_input_size = in EvalCalibration() local
|