Searched refs:aux_input_zp (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 306 const int32_t* aux_input_zp, const int8_t* aux_input_to_gate_weights, in CalculateLstmGateHybrid() argument
371 /*per_channel_scale=*/nullptr, aux_input_zp, accum_scratch, in CalculateLstmGateHybrid()
1032 float* output_ptr, int32_t* input_zp, int32_t* aux_input_zp, in LstmStepHybrid()
1130 aux_input_zp, asymmetric_quantize_inputs); in LstmStepHybrid()
1143 aux_input_zp, aux_input_to_input_weights_ptr, in LstmStepHybrid()
1160 aux_input_zp, aux_input_to_forget_weights_ptr, in LstmStepHybrid()
1176 aux_input_zp, aux_input_to_cell_weights_ptr, in LstmStepHybrid()
1197 aux_input_zp, aux_input_to_output_weights_ptr, in LstmStepHybrid()
1910 TfLiteTensor* input_zp, TfLiteTensor* aux_input_zp, in EvalHybrid()
1958 aux_input_zp_ptr = GetTensorData<int32_t>(aux_input_zp); in EvalHybrid()
|
| D | bidirectional_sequence_lstm.cc | 866 TfLiteTensor* aux_input_zp; in Prepare() local
869 GetTemporarySafe(context, node, kAuxInputZeroPoints, &aux_input_zp)); in Prepare()
870 aux_input_zp->type = kTfLiteFloat32; in Prepare()
871 aux_input_zp->allocation_type = kTfLiteArenaRw; in Prepare()
872 if (!TfLiteIntArrayEqualsArray(aux_input_zp->dims, 1, scaling_dims)) { in Prepare()
875 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, aux_input_zp, in Prepare()
|
| D | lstm_eval.h | 168 TfLiteTensor* input_zp, TfLiteTensor* aux_input_zp,
|