Home
last modified time | relevance | path

Searched refs:output_zp (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils.h126 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate() argument
130 n_output, output_zp, scratch, output, context); in MatrixBatchVectorMultiplyAccumulate()
136 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate() argument
140 n_output, output_zp, scratch, output, context); in MatrixBatchVectorMultiplyAccumulate()
167 int32_t output_zp, int8_t* proj_output) { in MatrixBatchVectorMultiply() argument
171 n_hidden, n_output, output_zp, proj_output); in MatrixBatchVectorMultiply()
220 int32_t n_input, int32_t output_zp, int8_t* output) { in CwiseMul() argument
222 output_zp, output); in CwiseMul()
Dportable_tensor_utils_impl.h111 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp,
117 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp,
133 int32_t n_output, int32_t output_zp, int8_t* proj_output);
170 int32_t n_input, int32_t output_zp, int8_t* output);
Dportable_tensor_utils.cc296 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in PortableMatrixBatchVectorMultiplyAccumulateImpl() argument
309 acc += output_zp; in PortableMatrixBatchVectorMultiplyAccumulateImpl()
325 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in PortableMatrixBatchVectorMultiplyAccumulate() argument
329 n_output, output_zp, output); in PortableMatrixBatchVectorMultiplyAccumulate()
335 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in PortableMatrixBatchVectorMultiplyAccumulate() argument
339 n_output, output_zp, output); in PortableMatrixBatchVectorMultiplyAccumulate()
378 int32_t n_output, int32_t output_zp, int8_t* proj_output) { in PortableMatrixBatchVectorMultiply() argument
398 acc += output_zp; in PortableMatrixBatchVectorMultiply()
623 int32_t n_input, int32_t output_zp, int8_t* output) { in PortableCwiseMul() argument
631 value -= output_zp; in PortableCwiseMul()
Dsvdf.h75 int32_t output_zp) { in EvalIntegerSVDF() argument
145 int32_t x3 = x2 + output_zp; in EvalIntegerSVDF()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dsse_tensor_utils.h102 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate() argument
106 shift, n_batch, n_input, n_output, output_zp, scratch, output, context); in MatrixBatchVectorMultiplyAccumulate()
112 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate() argument
116 shift, n_batch, n_input, n_output, output_zp, scratch, output, context); in MatrixBatchVectorMultiplyAccumulate()
137 int32_t output_zp, int8_t* proj_output) { in MatrixBatchVectorMultiply() argument
141 n_hidden, n_output, output_zp, proj_output); in MatrixBatchVectorMultiply()
196 int32_t n_input, int32_t output_zp, int8_t* output) { in CwiseMul() argument
198 output_zp, output); in CwiseMul()
Dneon_tensor_utils.h93 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate() argument
97 n_output, output_zp, scratch, output, context); in MatrixBatchVectorMultiplyAccumulate()
103 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in MatrixBatchVectorMultiplyAccumulate() argument
107 n_output, output_zp, scratch, output, context); in MatrixBatchVectorMultiplyAccumulate()
128 int32_t output_zp, int8_t* proj_output) { in MatrixBatchVectorMultiply() argument
132 n_hidden, n_output, output_zp, proj_output); in MatrixBatchVectorMultiply()
188 int32_t output_zp, int8_t* output) { in CwiseMul() argument
190 n_input, output_zp, output); in CwiseMul()
Dneon_tensor_utils_impl.h78 int32_t output_zp, int8_t* output);
93 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp,
99 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp,
Dneon_tensor_utils.cc744 int32_t n_output, int32_t output_zp, in NeonMatrixBatchVectorMultiplyImpl() argument
854 int32_t output_zp, int32_t* scratch, int16_t* output) { in NeonMatrixBatchVectorAccumulateImpl() argument
861 const int32x4_t output_zp_dup = vdupq_n_s32(output_zp); in NeonMatrixBatchVectorAccumulateImpl()
893 temp += output_zp; in NeonMatrixBatchVectorAccumulateImpl()
907 int32_t output_zp, int32_t* scratch, int8_t* output) { in NeonMatrixBatchVectorAccumulateImpl() argument
914 const int32x4_t output_zp_dup = vdupq_n_s32(output_zp); in NeonMatrixBatchVectorAccumulateImpl()
967 temp += output_zp; in NeonMatrixBatchVectorAccumulateImpl()
981 int32_t n_input, int32_t n_output, int32_t output_zp, in NeonCpuBackendGemm() argument
1014 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp, in NeonMatrixBatchVectorMultiplyAccumulate() argument
1018 n_output, output_zp, scratch, context); in NeonMatrixBatchVectorMultiplyAccumulate()
[all …]
Dsse_tensor_utils.cc178 int32_t n_input, int32_t n_output, int32_t output_zp, in SseCpuBackendGemm() argument
Doptimized_ops.h5783 int32_t output_zp, int32_t* scratch, uint8_t* output) { in Quantize() argument
5790 const int32x4_t output_zp_dup = vdupq_n_s32(output_zp); in Quantize()
5834 temp += output_zp; in Quantize()
5847 int32_t output_zp, int32_t output_min, int32_t output_max, in Quantize() argument
5862 const int32x4_t output_offset_vec = vdupq_n_s32(output_zp); in Quantize()
5928 acc += output_zp; in Quantize()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dreduce.cc43 int output_zp; member
118 op_data->output_zp = output->params.zero_point; in PrepareMeanOrSum()
188 tflite::micro::GetTensorData<int8_t>(output), op_data->output_zp); in EvalMean()
189 } else if (op_data->input_zp == op_data->output_zp && in EvalMean()
210 op_data->output_zp, op_data->output_scale, output->dims->data, in EvalMean()
224 op_data->output_zp, op_data->output_scale); in EvalMean()
225 } else if (op_data->input_zp == op_data->output_zp && in EvalMean()
247 op_data->output_zp, op_data->output_scale, output->dims->data, in EvalMean()
294 TF_LITE_ENSURE_EQ(context, op_data->input_zp, op_data->output_zp); in EvalMax()
Dreduce_test.cc550 int output_zp = 0; in TF_LITE_MICRO_TEST() local
561 output_data_quant, expected_output_data_quant, output_scale, output_zp, in TF_LITE_MICRO_TEST()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_utils.cc31 int64_t input_zp, int64_t output_zp, bool double_round) { in buildRescale() argument
41 rewriter.getI32IntegerAttr(static_cast<int32_t>(output_zp)), in buildRescale()
66 double output_scale, int64_t output_zp) { in buildRescaleFromInt32() argument
75 output_zp, true); in buildRescaleFromInt32()
90 int64_t output_zp = output_qtype.getZeroPoint(); in buildRescaleOpConvOutput() local
109 rewriter.getI32IntegerAttr(output_zp), in buildRescaleOpConvOutput()
130 int64_t output_zp = output_qtype.getZeroPoint(); in buildRescaleOpConvOutput() local
150 rewriter.getI32IntegerAttr(output_zp), in buildRescaleOpConvOutput()
Dlegalize_utils.h40 int64_t input_zp, int64_t output_zp,
51 double output_scale, int64_t output_zp);
Dlegalize_common.cc2313 int64_t input_zp, double output_scale, int64_t output_zp) { in convertReduceOpCommon() argument
2355 output_scale, output_zp); in convertReduceOpCommon()
2470 int64_t output_zp = 0; in convertReduceSumOp() local
2487 output_zp = output_qtype.getZeroPoint(); in convertReduceSumOp()
2494 output_zp); in convertReduceSumOp()
2541 int64_t output_zp = 0; in convertReduceMeanOp() local
2558 output_zp = output_qtype.getZeroPoint(); in convertReduceMeanOp()
2565 output_zp); in convertReduceMeanOp()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtensor_utils.h107 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp,
138 int32_t n_batch, int32_t n_input, int32_t n_output, int32_t output_zp,
Dtensor_utils_common.h174 int32_t output_zp, int8_t* proj_output);
286 int32_t n_input, int32_t output_zp, int8_t* output);
Dtensor_utils_test.cc549 const int32_t output_zp = -11; in TEST() local
557 /*n_batch=*/2, /*n_input=*/30, /*n_output=*/9, output_zp, scratch.data(), in TEST()
598 const int32_t output_zp = -11; in TEST() local
605 /*n_batch=*/2, /*n_input=*/30, /*n_cell=*/9, output.data(), output_zp); in TEST()
647 const int32_t output_zp = -11; in TEST() local
655 /*n_batch=*/2, /*n_hidden=*/30, /*n_output=*/9, output_zp, output.data()); in TEST()
/external/llvm-project/mlir/include/mlir/Dialect/Tosa/IR/
DTosaOpBase.td64 StructFieldAttr<"output_zp", I32Attr>
DTosaOps.td1484 I32Attr:$output_zp,
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_model.cc471 const float output_zp = output_tensor->quantization->zero_point[0]; in ApplyConstraints() local
482 input_tensor->quantization->zero_point[0] == output_zp) { in ApplyConstraints()
492 output_zp, &additional_tensor); in ApplyConstraints()
/external/llvm-project/mlir/test/Dialect/Tosa/
Dops.mlir446 …e_round = false, input_zp = 127 : i32, multiplier = [1073741824 : i32], output_zp = -1 : i32, per_…
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md3264 …stride_w}, padding=tosa_padding, quantization_info={input_zp=%input.zp, output_zp=%output.zp}} : (…