Searched defs:aux_input_size (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| 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()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| 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::__anon7adf078d0111::BidirectionalRNNOpModel
1421 void run_closedbox_test_with_input_split(int input_size, int aux_input_size) { in run_closedbox_test_with_input_split()
|
| D | bidirectional_sequence_rnn.cc | 391 const int aux_input_size = (aux_input) ? aux_input->dims->data[2] : 0; in EvalFloat() local
537 const int aux_input_size = (aux_input) ? aux_input->dims->data[2] : 0; in EvalHybrid() local
|
| D | lstm_eval.cc | 1724 const int aux_input_size = in EvalFloat() local
1923 const int aux_input_size = in EvalHybrid() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 371 const int aux_input_size = in EvalFloat() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 302 const int aux_input_size = in EvalCalibration() local
|