| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | reduce.cc | 41 int input_zp; member
116 op_data->input_zp = input->params.zero_point; in PrepareMeanOrSum()
186 tflite::micro::GetTensorData<int8_t>(input), op_data->input_zp, in EvalMean()
189 } else if (op_data->input_zp == op_data->output_zp && in EvalMean()
207 tflite::micro::GetTensorData<int8_t>(input), op_data->input_zp, in EvalMean()
221 op_data->input_zp, op_data->input_scale, in EvalMean()
225 } else if (op_data->input_zp == op_data->output_zp && in EvalMean()
244 tflite::micro::GetTensorData<uint8_t>(input), op_data->input_zp, in EvalMean()
294 TF_LITE_ENSURE_EQ(context, op_data->input_zp, op_data->output_zp); in EvalMax()
|
| D | reduce_test.cc | 524 int input_zp = 0; in TF_LITE_MICRO_TEST() local
533 input_shape, input_data, input_data_quant, input_scale, input_zp, in TF_LITE_MICRO_TEST()
535 output_data_quant, expected_output_data_quant, input_scale, input_zp, in TF_LITE_MICRO_TEST()
548 int input_zp = 0; in TF_LITE_MICRO_TEST() local
559 input_shape, input_data, input_data_quant, input_scale, input_zp, in TF_LITE_MICRO_TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_utils.h | 40 int64_t input_zp, int64_t output_zp,
46 int64_t input_zp);
|
| D | legalize_utils.cc | 31 int64_t input_zp, int64_t output_zp, bool double_round) { in buildRescale() argument
40 rewriter.getI32IntegerAttr(static_cast<int32_t>(input_zp)), in buildRescale()
52 int64_t input_zp) { in buildRescaleToInt32() argument
60 input_zp, 0, false); in buildRescaleToInt32()
|
| D | legalize_common.cc | 2313 int64_t input_zp, double output_scale, int64_t output_zp) { in convertReduceOpCommon() argument
2333 val = buildRescaleToInt32(rewriter, op, val, input_scale, input_zp); in convertReduceOpCommon()
2469 int64_t input_zp = 0; in convertReduceSumOp() local
2486 input_zp = input_qtype.getZeroPoint(); in convertReduceSumOp()
2493 reduce_element_type, input_is_qtype, input_scale, input_zp, output_scale, in convertReduceSumOp()
2540 int64_t input_zp = 0; in convertReduceMeanOp() local
2557 input_zp = input_qtype.getZeroPoint(); in convertReduceMeanOp()
2564 reduce_element_type, input_is_qtype, input_scale, input_zp, output_scale, in convertReduceMeanOp()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 300 const int8_t* input, const float* input_sf, const int32_t* input_zp, in CalculateLstmGateHybrid() argument
361 /*per_channel_scale=*/nullptr, input_zp, accum_scratch, in CalculateLstmGateHybrid()
653 const int8_t* input, int32_t input_zp, const int8_t* input_to_gate_weight, in CalculateLstmGateInteger8x8_8() argument
679 input, input_zp, input_to_gate_weight, input_to_gate_scale_a, in CalculateLstmGateInteger8x8_8()
1032 float* output_ptr, int32_t* input_zp, int32_t* aux_input_zp, 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 …]
|
| D | bidirectional_sequence_lstm.cc | 853 TfLiteTensor* input_zp; in Prepare() local
855 context, GetTemporarySafe(context, node, kInputZeroPoints, &input_zp)); in Prepare()
856 input_zp->type = kTfLiteFloat32; in Prepare()
857 input_zp->allocation_type = kTfLiteArenaRw; in Prepare()
858 if (!TfLiteIntArrayEqualsArray(input_zp->dims, 1, scaling_dims)) { in Prepare()
862 context, context->ResizeTensor(context, input_zp, input_zp_size)); in Prepare()
|
| D | unidirectional_sequence_lstm.cc | 1112 TfLiteTensor* input_zp; in Prepare() local
1114 context, GetTemporarySafe(context, node, kInputZeroPoints, &input_zp)); in Prepare()
1115 input_zp->type = kTfLiteFloat32; in Prepare()
1116 input_zp->allocation_type = kTfLiteArenaRw; in Prepare()
1117 if (!TfLiteIntArrayEqualsArray(input_zp->dims, 1, scaling_dims)) { in Prepare()
1121 context, context->ResizeTensor(context, input_zp, input_zp_size)); in Prepare()
|
| D | lstm_eval.h | 168 TfLiteTensor* input_zp, TfLiteTensor* aux_input_zp,
|
| D | lstm.cc | 606 int input_zp = 0; in PopulateQuantizedLstmParams8x8_8() local
669 input_zp = input->params.zero_point; in PopulateQuantizedLstmParams8x8_8()
1570 TfLiteTensor* input_zp; in Prepare() local
1572 context, GetTemporarySafe(context, node, kInputZeroPoints, &input_zp)); in Prepare()
1573 input_zp->type = kTfLiteFloat32; in Prepare()
1574 input_zp->allocation_type = kTfLiteArenaRw; in Prepare()
1575 if (!TfLiteIntArrayEqualsArray(input_zp->dims, 1, scaling_dims)) { in Prepare()
1579 context, context->ResizeTensor(context, input_zp, input_zp_size)); in Prepare()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Tosa/IR/ |
| D | TosaOpBase.td | 63 StructFieldAttr<"input_zp", I32Attr>,
82 StructFieldAttr<"input_zp", I32Attr>,
102 StructFieldAttr<"input_zp", I32Attr>
|
| D | TosaOps.td | 1483 I32Attr:$input_zp,
|
| /external/llvm-project/mlir/test/Dialect/Tosa/ |
| D | quant-test.mlir | 15 … [1, 1, 2, 2], dilation = [2, 1], stride = [1, 1], quantization_info = {input_zp = -1 : i32, weigh…
|
| D | ops.mlir | 446 …%0 = "tosa.rescale"(%arg0) {double_round = false, input_zp = 127 : i32, multiplier = [1073741824 :…
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | svdf.h | 74 int scale_1_b, int32_t scale_2_a, int scale_2_b, int32_t input_zp, in EvalIntegerSVDF() argument
101 dot_prod += *matrix_data++ * (*vector_in_batch++ - input_zp); in EvalIntegerSVDF()
|
| D | portable_tensor_utils.h | 303 void TwoGateSaturatingAdd(const int8_t* input, int8_t input_zp, in TwoGateSaturatingAdd() argument
311 input, input_zp, recurrent, recurrent_zp, input_effective_scale_a, in TwoGateSaturatingAdd()
|
| D | portable_tensor_utils_impl.h | 222 void PortableTwoGateSaturatingAdd(const int8_t* input, int8_t input_zp,
|
| D | portable_tensor_utils.cc | 744 void PortableTwoGateSaturatingAdd(const int8_t* input, int8_t input_zp, in PortableTwoGateSaturatingAdd() argument
755 int32_t x = static_cast<int32_t>(input[i]) - static_cast<int32_t>(input_zp); in PortableTwoGateSaturatingAdd()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | neon_tensor_utils.h | 300 void TwoGateSaturatingAdd(const int8_t* input, int8_t input_zp, in TwoGateSaturatingAdd() argument
308 input, input_zp, recurrent, recurrent_zp, input_effective_scale_a, in TwoGateSaturatingAdd()
|
| D | sse_tensor_utils.h | 309 void TwoGateSaturatingAdd(const int8_t* input, int8_t input_zp, in TwoGateSaturatingAdd() argument
317 input, input_zp, recurrent, recurrent_zp, input_effective_scale_a, in TwoGateSaturatingAdd()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | tensor_utils_common.h | 455 void TwoGateSaturatingAdd(const int8_t* input, int8_t input_zp,
|
| D | tensor_utils_test.cc | 569 const int32_t input_zp = 3; in TEST() local
604 input.data(), input_zp, input_to_gate_weights.data(), multiplier, shift, in TEST()
1974 void TwoGateSaturatingAdd(const int8_t* input, int8_t input_zp,
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | ceva_tflm_lib.h | 261 int16_t* result_in_batch, int input_zp, int n_input,
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 3264 …, stride={stride_h, stride_w}, padding=tosa_padding, quantization_info={input_zp=%input.zp, output…
3392 …w}, dilation={dilation_h_factor, dilation_w_factor}, quantization_info={input_zp=%input.zp, weight…
3472 …w}, dilation={dilation_h_factor, dilation_w_factor}, quantization_info={input_zp=%input.zp, weight…
|