Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.cc300 const int8_t* input, const float* input_sf, const int32_t* input_zp, in CalculateLstmGateHybrid() argument
351 scales[i] = input_to_gate_weights_scale * input_sf[i]; in CalculateLstmGateHybrid()
360 input_to_gate_weights_scale, input_sf, n_batch, gate, in CalculateLstmGateHybrid()
380 scales[i] = recurrent_to_gate_weights_scale * input_sf[i]; in CalculateLstmGateHybrid()
1027 float* input_sf, float* aux_input_sf, float* output_state_sf, in LstmStepHybrid()
1124 quantized_input_ptr, input_sf, input_zp, in LstmStepHybrid()
1140 quantized_input_ptr, input_sf, input_zp, input_to_input_weights_ptr, in LstmStepHybrid()
1157 quantized_input_ptr, input_sf, input_zp, input_to_forget_weights_ptr, in LstmStepHybrid()
1173 quantized_input_ptr, input_sf, input_zp, input_to_cell_weights_ptr, in LstmStepHybrid()
1194 quantized_input_ptr, input_sf, input_zp, input_to_output_weights_ptr, in LstmStepHybrid()
[all …]
Dbidirectional_sequence_lstm.cc748 TfLiteTensor* input_sf; in Prepare() local
751 GetTemporarySafe(context, node, kInputScalingFactors, &input_sf)); in Prepare()
752 input_sf->type = kTfLiteFloat32; in Prepare()
753 input_sf->allocation_type = kTfLiteArenaRw; in Prepare()
755 if (!TfLiteIntArrayEqualsArray(input_sf->dims, 1, scaling_dims)) { in Prepare()
759 context, context->ResizeTensor(context, input_sf, input_sf_size)); in Prepare()
Dunidirectional_sequence_lstm.cc1029 TfLiteTensor* input_sf; in Prepare() local
1032 GetTemporarySafe(context, node, kInputScalingFactors, &input_sf)); in Prepare()
1033 input_sf->type = kTfLiteFloat32; in Prepare()
1034 input_sf->allocation_type = kTfLiteArenaRw; in Prepare()
1036 if (!TfLiteIntArrayEqualsArray(input_sf->dims, 1, scaling_dims)) { in Prepare()
1040 context, context->ResizeTensor(context, input_sf, input_sf_size)); in Prepare()
Dlstm_eval.h161 TfLiteTensor* scratch_buffer, TfLiteTensor* input_sf,
Dlstm.cc1487 TfLiteTensor* input_sf; in Prepare() local
1490 GetTemporarySafe(context, node, kInputScalingFactors, &input_sf)); in Prepare()
1491 input_sf->type = kTfLiteFloat32; in Prepare()
1492 input_sf->allocation_type = kTfLiteArenaRw; in Prepare()
1494 if (!TfLiteIntArrayEqualsArray(input_sf->dims, 1, scaling_dims)) { in Prepare()
1498 context, context->ResizeTensor(context, input_sf, input_sf_size)); in Prepare()