Home
last modified time | relevance | path

Searched refs:aux_input_quantized (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dbidirectional_sequence_rnn.cc330 TfLiteTensor* aux_input_quantized; in Prepare() local
333 &aux_input_quantized)); in Prepare()
334 aux_input_quantized->type = fw_input_weights->type; in Prepare()
335 aux_input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
336 if (!TfLiteIntArrayEqual(aux_input_quantized->dims, aux_input->dims)) { in Prepare()
340 context->ResizeTensor(context, aux_input_quantized, in Prepare()
524 TfLiteTensor* aux_input_quantized, TfLiteTensor* fw_hidden_state_quantized, in EvalHybrid() argument
566 aux_quantized_input_ptr = GetTensorData<int8_t>(aux_input_quantized); in EvalHybrid()
818 TfLiteTensor* aux_input_quantized = in Eval() local
826 input_quantized, aux_input_quantized, fw_hidden_state_quantized, in Eval()
Dbidirectional_sequence_lstm.cc950 TfLiteTensor* aux_input_quantized; in Prepare() local
953 &aux_input_quantized)); in Prepare()
954 aux_input_quantized->type = fw_input_to_output_weights->type; in Prepare()
955 aux_input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
956 if (!TfLiteIntArrayEqual(aux_input_quantized->dims, aux_input->dims)) { in Prepare()
960 context->ResizeTensor(context, aux_input_quantized, in Prepare()
1247 TfLiteTensor* aux_input_quantized = in Eval() local
1293 aux_input_quantized, fw_activation_state_quantized, in Eval()
1334 aux_input_quantized, bw_activation_state_quantized, in Eval()
Dlstm_eval.h164 TfLiteTensor* input_quantized, TfLiteTensor* aux_input_quantized,
Dlstm_eval.cc1906 TfLiteTensor* input_quantized, TfLiteTensor* aux_input_quantized, in EvalHybrid()
2037 GetTensorData<int8_t>(aux_input_quantized), in EvalHybrid()
2136 GetTensorData<int8_t>(aux_input_quantized), in EvalHybrid()