Home
last modified time | relevance | path

Searched refs:aux_input_sf (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.cc305 const int8_t* aux_input, const float* aux_input_sf, in CalculateLstmGateHybrid() argument
370 aux_input_to_gate_weights_scale, aux_input_sf, n_batch, gate, in CalculateLstmGateHybrid()
1027 float* input_sf, float* aux_input_sf, float* output_state_sf, in LstmStepHybrid()
1129 quantized_aux_input_ptr, aux_input_sf, in LstmStepHybrid()
1142 input_to_input_row_sums, quantized_aux_input_ptr, aux_input_sf, in LstmStepHybrid()
1159 input_to_forget_row_sums, quantized_aux_input_ptr, aux_input_sf, in LstmStepHybrid()
1175 input_to_cell_row_sums, quantized_aux_input_ptr, aux_input_sf, in LstmStepHybrid()
1196 input_to_output_row_sums, quantized_aux_input_ptr, aux_input_sf, in LstmStepHybrid()
1904 TfLiteTensor* aux_input_sf, TfLiteTensor* output_state_sf, in EvalHybrid()
2032 GetTensorData<float>(aux_input_sf), in EvalHybrid()
[all …]
Dbidirectional_sequence_lstm.cc763 TfLiteTensor* aux_input_sf; in Prepare() local
766 &aux_input_sf)); in Prepare()
767 aux_input_sf->type = kTfLiteFloat32; in Prepare()
768 aux_input_sf->allocation_type = kTfLiteArenaRw; in Prepare()
769 if (!TfLiteIntArrayEqualsArray(aux_input_sf->dims, 1, scaling_dims)) { in Prepare()
772 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, aux_input_sf, in Prepare()
Dlstm_eval.h162 TfLiteTensor* aux_input_sf, TfLiteTensor* output_state_sf,