Searched refs:output_state_zp (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 311 const int32_t* output_state_zp, const int8_t* recurrent_to_gate_weights, in CalculateLstmGateHybrid() argument
389 /*per_channel_scale=*/nullptr, output_state_zp, accum_scratch, in CalculateLstmGateHybrid()
622 int32_t output_state_zp, int8_t quantized_proj_clip, int8_t* output_state, in CalculateLstmOutputInteger8x8_16() argument
638 proj_scale_b, n_batch, n_cell, n_output, output_state_zp, scratch2, in CalculateLstmOutputInteger8x8_16()
659 const int8_t* output_state, const int32_t output_state_zp, in CalculateLstmGateInteger8x8_8() argument
684 output_state, output_state_zp, recurrent_to_gate_weight, in CalculateLstmGateInteger8x8_8()
728 int32_t output_state_zp, int32_t quantized_proj_clip, int8_t* output_state, in CalculateLstmOutputInteger8x8_8() argument
737 n_batch, n_cell, n_output, output_state_zp, output_state); in CalculateLstmOutputInteger8x8_8()
1033 int32_t* output_state_zp, int32_t* row_sums, int row_sums_size, in LstmStepHybrid()
1135 output_state_sf, output_state_zp, asymmetric_quantize_inputs); in LstmStepHybrid()
[all …]
|
| D | bidirectional_sequence_lstm.cc | 880 TfLiteTensor* output_state_zp; in Prepare() local
883 &output_state_zp)); in Prepare()
884 output_state_zp->type = kTfLiteFloat32; in Prepare()
885 output_state_zp->allocation_type = kTfLiteArenaRw; in Prepare()
886 if (!TfLiteIntArrayEqualsArray(output_state_zp->dims, 1, scaling_dims)) { in Prepare()
889 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, output_state_zp, in Prepare()
|
| D | unidirectional_sequence_lstm.cc | 1125 TfLiteTensor* output_state_zp; in Prepare() local
1128 &output_state_zp)); in Prepare()
1129 output_state_zp->type = kTfLiteFloat32; in Prepare()
1130 output_state_zp->allocation_type = kTfLiteArenaRw; in Prepare()
1131 if (!TfLiteIntArrayEqualsArray(output_state_zp->dims, 1, scaling_dims)) { in Prepare()
1134 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, output_state_zp, in Prepare()
|
| D | lstm_eval.h | 169 TfLiteTensor* output_state_zp, TfLiteTensor* row_sums, int row_sums_size,
|
| D | lstm.cc | 607 int output_state_zp = 0; in PopulateQuantizedLstmParams8x8_8() local
670 output_state_zp = output_state->params.zero_point; in PopulateQuantizedLstmParams8x8_8()
1583 TfLiteTensor* output_state_zp; in Prepare() local
1586 &output_state_zp)); in Prepare()
1587 output_state_zp->type = kTfLiteFloat32; in Prepare()
1588 output_state_zp->allocation_type = kTfLiteArenaRw; in Prepare()
1589 if (!TfLiteIntArrayEqualsArray(output_state_zp->dims, 1, scaling_dims)) { in Prepare()
1592 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, output_state_zp, in Prepare()
|