| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | kernel_utils.cc | 42 int input_size, int aux_input_size, int num_units, in RnnBatchStep() argument
59 if (aux_input_size > 0) { in RnnBatchStep()
61 aux_input_weights_ptr, num_units, aux_input_size, aux_input_ptr_batch, in RnnBatchStep()
91 if (aux_input_size > 0) { in RnnBatchStep()
94 aux_input_weights_ptr, num_units, aux_input_size, in RnnBatchStep()
95 aux_input_ptr_batch + k * aux_input_size, in RnnBatchStep()
148 const float* bias_ptr, int input_size, int aux_input_size, int num_units, in RnnBatchStep() argument
175 aux_input_size); in RnnBatchStep()
208 batch_size * aux_input_size)) { in RnnBatchStep()
210 aux_input_ptr_batch, batch_size, aux_input_size, in RnnBatchStep()
[all …]
|
| D | kernel_utils.h | 51 int input_size, int aux_input_size, int num_units,
82 const float* bias_ptr, int input_size, int aux_input_size, int num_units,
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_rnn.cc | 391 const int aux_input_size = (aux_input) ? aux_input->dims->data[2] : 0; in EvalFloat() local
434 input_size, aux_input_size, fw_num_units, batch_size, fw_output_step, in EvalFloat()
455 input_size, aux_input_size, bw_num_units, batch_size, bw_output_step, in EvalFloat()
472 b * aux_input_size * max_time + s * aux_input_size in EvalFloat()
479 input_size, aux_input_size, fw_num_units, /*batch_size=*/1, in EvalFloat()
498 b * aux_input_size * max_time + s * aux_input_size in EvalFloat()
505 input_size, aux_input_size, bw_num_units, /*batch_size=*/1, in EvalFloat()
537 const int aux_input_size = (aux_input) ? aux_input->dims->data[2] : 0; in EvalHybrid() local
561 if (aux_input_size > 0) { in EvalHybrid()
608 fw_recurrent_weights_scale, fw_bias_ptr, input_size, aux_input_size, in EvalHybrid()
[all …]
|
| D | bidirectional_sequence_rnn_test.cc | 666 int bw_units, int input_size, int aux_input_size, in BidirectionalRNNOpModel() argument
675 aux_input_size_(aux_input_size), in BidirectionalRNNOpModel()
817 int aux_input_size() { return aux_input_size_; } in aux_input_size() function in tflite::__anon048612660111::BidirectionalRNNOpModel
1421 void run_closedbox_test_with_input_split(int input_size, int aux_input_size) { in run_closedbox_test_with_input_split() argument
1425 input_size, aux_input_size, in run_closedbox_test_with_input_split()
1430 std::vector<float> aux_weights(num_units * rnn.aux_input_size()); in run_closedbox_test_with_input_split()
1445 weights.begin() + weights_offset + rnn.aux_input_size(), in run_closedbox_test_with_input_split()
1447 weights_offset += rnn.aux_input_size(); in run_closedbox_test_with_input_split()
1448 aux_weights_offset += rnn.aux_input_size(); in run_closedbox_test_with_input_split()
1461 (rnn.input_size() + rnn.aux_input_size()) * rnn.sequence_len(); in run_closedbox_test_with_input_split()
[all …]
|
| D | lstm_eval.cc | 1724 const int aux_input_size = in EvalFloat() local
1796 n_input, aux_input_size, n_output, output_batch_leading_dim, in EvalFloat()
1857 n_cell, n_input, aux_input_size, n_output, output_batch_leading_dim, in EvalFloat()
1923 const int aux_input_size = in EvalHybrid() local
2029 n_input, aux_input_size, n_output, output_batch_leading_dim, in EvalHybrid()
2127 /*n_batch=*/1, n_cell, n_input, aux_input_size, n_output, in EvalHybrid()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 302 const int aux_input_size = in EvalCalibration() local
374 n_input, aux_input_size, n_output, output_batch_leading_dim, in EvalCalibration()
436 n_cell, n_input, aux_input_size, n_output, output_batch_leading_dim, in EvalCalibration()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 371 const int aux_input_size = in EvalFloat() local
443 n_input, aux_input_size, n_output, output_batch_leading_dim, in EvalFloat()
505 n_cell, n_input, aux_input_size, n_output, output_batch_leading_dim, in EvalFloat()
|