Home
last modified time | relevance | path

Searched refs:aux_input_size (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dkernel_utils.cc40 int input_size, int aux_input_size, int num_units, in RnnBatchStep() argument
57 if (aux_input_size > 0) { in RnnBatchStep()
59 aux_input_weights_ptr, num_units, aux_input_size, aux_input_ptr_batch, in RnnBatchStep()
89 if (aux_input_size > 0) { in RnnBatchStep()
92 aux_input_weights_ptr, num_units, aux_input_size, in RnnBatchStep()
93 aux_input_ptr_batch + k * aux_input_size, in RnnBatchStep()
145 const float* bias_ptr, int input_size, int aux_input_size, int num_units, in RnnBatchStep() argument
182 batch_size * aux_input_size)) { in RnnBatchStep()
185 const int offset = b * aux_input_size; in RnnBatchStep()
187 aux_input_ptr_batch + offset, aux_input_size, in RnnBatchStep()
[all …]
Dkernel_utils.h51 int input_size, int aux_input_size, int num_units,
80 const float* bias_ptr, int input_size, int aux_input_size, int num_units,
/external/tensorflow/tensorflow/lite/kernels/
Dbidirectional_sequence_rnn.cc301 const int aux_input_size = (aux_input) ? aux_input->dims->data[2] : 0; in EvalFloat() local
340 input_size, aux_input_size, fw_num_units, batch_size, fw_output_step, in EvalFloat()
360 input_size, aux_input_size, bw_num_units, batch_size, bw_output_step, in EvalFloat()
382 input_size, aux_input_size, fw_num_units, /*batch_size=*/1, in EvalFloat()
405 input_size, aux_input_size, bw_num_units, /*batch_size=*/1, in EvalFloat()
435 const int aux_input_size = (aux_input) ? aux_input->dims->data[2] : 0; in EvalHybrid() local
461 if (aux_input_size > 0) { in EvalHybrid()
503 fw_recurrent_weights_scale, fw_bias_ptr, input_size, aux_input_size, in EvalHybrid()
527 bw_recurrent_weights_scale, bw_bias_ptr, input_size, aux_input_size, in EvalHybrid()
554 fw_recurrent_weights_scale, fw_bias_ptr, input_size, aux_input_size, in EvalHybrid()
[all …]
Dlstm_eval.cc908 const int aux_input_size = in EvalFloat() local
969 if (aux_input_size > 0) { in EvalFloat()
1009 params, n_batch, n_cell, n_input, aux_input_size, n_output, in EvalFloat()
1058 aux_input_size, n_output, output_batch_leading_dim, in EvalFloat()
1112 const int aux_input_size = in EvalHybrid() local
1247 if (aux_input_size > 0) { in EvalHybrid()
1308 n_cell, n_input, aux_input_size, n_output, output_batch_leading_dim, in EvalHybrid()
1366 /*n_batch=*/1, n_cell, n_input, aux_input_size, n_output, in EvalHybrid()
Dbidirectional_sequence_rnn_test.cc674 int aux_input_size = 0; in BidirectionalRNNOpModel() local
677 aux_input_size = input_size_; in BidirectionalRNNOpModel()
709 {batches_, sequence_len_, aux_input_size}, // aux_input in BidirectionalRNNOpModel()