Home
last modified time | relevance | path

Searched refs:aux_input (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dbidirectional_sequence_rnn.cc129 const TfLiteTensor* aux_input = in Prepare() local
176 TF_LITE_ASSERT_EQ(aux_input->dims->data[0], input->dims->data[0]); in Prepare()
177 TF_LITE_ASSERT_EQ(aux_input->dims->data[1], input->dims->data[1]); in Prepare()
182 TF_LITE_ASSERT_EQ(aux_input->dims->data[2], in Prepare()
184 TF_LITE_ASSERT_EQ(aux_input->dims->data[2], in Prepare()
336 if (!TfLiteIntArrayEqual(aux_input_quantized->dims, aux_input->dims)) { in Prepare()
338 TfLiteIntArrayCopy(aux_input->dims); in Prepare()
379 const TfLiteTensor* aux_input, in EvalFloat() argument
391 const int aux_input_size = (aux_input) ? aux_input->dims->data[2] : 0; in EvalFloat()
425 (aux_input != nullptr) in EvalFloat()
[all …]
Dlstm_eval.cc152 const float* aux_input, const float* aux_input_to_gate_weights, in CalculateLstmGateFloat() argument
181 aux_input, n_batch, gate); in CalculateLstmGateFloat()
305 const int8_t* aux_input, const float* aux_input_sf, in CalculateLstmGateHybrid() argument
369 aux_input_to_gate_weights, n_cell, n_aux_input, aux_input, in CalculateLstmGateHybrid()
1703 const TfLiteTensor* aux_input, in EvalFloat() argument
1725 (aux_input) ? aux_input->dims->data[aux_input->dims->size - 1] : 0; in EvalFloat()
1764 if (aux_input) { in EvalFloat()
1765 aux_input_ptr = GetTensorData<float>(aux_input) + t_rel * input_step; in EvalFloat()
1813 if (aux_input) { in EvalFloat()
1815 GetTensorData<float>(aux_input) + time_offset * input_step; in EvalFloat()
[all …]
Dlstm_eval.h114 const TfLiteTensor* aux_input,
150 const TfLiteTensor* aux_input,
Dbidirectional_sequence_lstm.cc489 const TfLiteTensor* aux_input = in Prepare() local
527 TF_LITE_ASSERT_EQ(aux_input->dims->data[0], input->dims->data[0]); in Prepare()
528 TF_LITE_ASSERT_EQ(aux_input->dims->data[1], input->dims->data[1]); in Prepare()
956 if (!TfLiteIntArrayEqual(aux_input_quantized->dims, aux_input->dims)) { in Prepare()
958 TfLiteIntArrayCopy(aux_input->dims); in Prepare()
1121 const TfLiteTensor* aux_input = in Eval() local
1140 const bool has_previous_bw_output = (aux_input != nullptr); in Eval()
1170 const TfLiteTensor* bw_input = non_stacking_mode ? aux_input : input; in Eval()
1171 const TfLiteTensor* real_aux_input = non_stacking_mode ? nullptr : aux_input; in Eval()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc42 const float* aux_input, const float* aux_input_to_gate_weights, in CalculateLstmGateFloat() argument
73 aux_input, n_batch, gate); in CalculateLstmGateFloat()
279 const TfLiteTensor* aux_input, in EvalCalibration() argument
303 (aux_input) ? aux_input->dims->data[aux_input->dims->size - 1] : 0; in EvalCalibration()
342 if (aux_input) { in EvalCalibration()
343 aux_input_ptr = GetTensorData<float>(aux_input) + t_rel * input_step; in EvalCalibration()
392 if (aux_input) { in EvalCalibration()
394 GetTensorData<float>(aux_input) + time_offset * input_step; in EvalCalibration()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc347 const TfLiteTensor* aux_input, in EvalFloat() argument
372 (aux_input) ? aux_input->dims->data[aux_input->dims->size - 1] : 0; in EvalFloat()
411 if (aux_input) { in EvalFloat()
412 aux_input_ptr = GetTensorData<float>(aux_input) + t_rel * input_step; in EvalFloat()
461 if (aux_input) { in EvalFloat()
463 GetTensorData<float>(aux_input) + time_offset * input_step; in EvalFloat()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td4179 TFL_TensorOfOrNone<[F32, I8]>:$aux_input,