Home
last modified time | relevance | path

Searched refs:affine_quantization (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Ddepthwise_conv.cc182 const auto* affine_quantization = in Prepare() local
185 TF_LITE_ENSURE(context, affine_quantization); in Prepare()
186 TF_LITE_ENSURE(context, affine_quantization->scale); in Prepare()
187 TF_LITE_ENSURE(context, (affine_quantization->scale->size == 1 || in Prepare()
188 affine_quantization->scale->size == channels_out)); in Prepare()
202 const auto* affine_quantization = in Prepare() local
205 TF_LITE_ENSURE(context, affine_quantization); in Prepare()
206 TF_LITE_ENSURE(context, affine_quantization->scale); in Prepare()
208 context, affine_quantization->scale->size, in Prepare()
209 filter->dims->data[affine_quantization->quantized_dimension]); in Prepare()
[all …]
Dkernel_util.cc203 const auto* affine_quantization = in PopulateConvolutionQuantizationParams() local
208 per_channel_shift, affine_quantization->scale->size); in PopulateConvolutionQuantizationParams()
230 const auto* affine_quantization = in PopulateConvolutionQuantizationParams() local
232 TF_LITE_ENSURE(context, affine_quantization); in PopulateConvolutionQuantizationParams()
233 TF_LITE_ENSURE(context, affine_quantization->scale); in PopulateConvolutionQuantizationParams()
234 const bool is_per_channel = affine_quantization->scale->size > 1; in PopulateConvolutionQuantizationParams()
240 TF_LITE_ENSURE_EQ(context, affine_quantization->scale->size, num_channels); in PopulateConvolutionQuantizationParams()
243 filter->dims->data[affine_quantization->quantized_dimension]); in PopulateConvolutionQuantizationParams()
249 const float* filter_scales = affine_quantization->scale->data; in PopulateConvolutionQuantizationParams()
Dconv.cc375 const auto* affine_quantization = in Prepare() local
378 for (int i = 0; i < affine_quantization->zero_point->size; ++i) { in Prepare()
379 TF_LITE_ENSURE_EQ(context, affine_quantization->zero_point->data[i], 0); in Prepare()
415 const auto* affine_quantization = in Prepare() local
418 const float scale = affine_quantization->scale->data[0]; in Prepare()
419 for (int i = 1; i < affine_quantization->scale->size; i++) { in Prepare()
420 if (affine_quantization->scale->data[i] != scale) { in Prepare()
471 const auto* affine_quantization = in Prepare() local
474 TF_LITE_ENSURE(context, affine_quantization); in Prepare()
475 TF_LITE_ENSURE(context, affine_quantization->scale); in Prepare()
[all …]
Delementwise_test.cc77 auto* affine_quantization = in ExtractDequantVector() local
79 float scaling_factor = affine_quantization->scale->data[0]; in ExtractDequantVector()
80 int zero_point = affine_quantization->zero_point->data[0]; in ExtractDequantVector()
Dtest_util.h878 auto* affine_quantization = reinterpret_cast<TfLiteAffineQuantization*>( in QuantizeTensor() local
880 affine_quantization->quantized_dimension = 0; in QuantizeTensor()
881 affine_quantization->scale = TfLiteFloatArrayCreate(1); in QuantizeTensor()
882 affine_quantization->zero_point = TfLiteIntArrayCreate(1); in QuantizeTensor()
883 affine_quantization->scale->data[0] = scaling_factor; in QuantizeTensor()
884 affine_quantization->zero_point->data[0] = 0; in QuantizeTensor()
885 t->quantization.params = affine_quantization; in QuantizeTensor()
Dtranspose_conv.cc373 const auto* affine_quantization = in Prepare() local
377 TF_LITE_ENSURE(context, affine_quantization); in Prepare()
378 TF_LITE_ENSURE(context, affine_quantization->scale); in Prepare()
379 TF_LITE_ENSURE(context, (affine_quantization->scale->size == 1 || in Prepare()
380 affine_quantization->scale->size == channels_out)); in Prepare()
Dfully_connected.cc279 const auto* affine_quantization = in PrepareImpl() local
282 TF_LITE_ENSURE(context, affine_quantization); in PrepareImpl()
283 TF_LITE_ENSURE(context, affine_quantization->scale); in PrepareImpl()
284 const int per_channel_quantization_size = affine_quantization->scale->size; in PrepareImpl()
291 TF_LITE_ENSURE_EQ(context, affine_quantization->scale->size, in PrepareImpl()
295 filter->dims->data[affine_quantization->quantized_dimension]); in PrepareImpl()
299 const float* filter_scales = affine_quantization->scale->data; in PrepareImpl()
/external/tensorflow/tensorflow/lite/
Dinterpreter.cc66 auto* affine_quantization = reinterpret_cast<TfLiteAffineQuantization*>( in GetQuantizationFromLegacy() local
68 affine_quantization->scale = TfLiteFloatArrayCreate(1); in GetQuantizationFromLegacy()
69 affine_quantization->zero_point = TfLiteIntArrayCreate(1); in GetQuantizationFromLegacy()
70 affine_quantization->scale->data[0] = legacy_quantization.scale; in GetQuantizationFromLegacy()
71 affine_quantization->zero_point->data[0] = legacy_quantization.zero_point; in GetQuantizationFromLegacy()
72 quantization.params = affine_quantization; in GetQuantizationFromLegacy()
Dinterpreter_builder.cc439 auto* affine_quantization = reinterpret_cast<TfLiteAffineQuantization*>( in ParseQuantization() local
441 affine_quantization->scale = TfLiteFloatArrayCreate(num_scales); in ParseQuantization()
442 affine_quantization->zero_point = TfLiteIntArrayCreate(num_scales); in ParseQuantization()
444 affine_quantization->scale->data[i] = src_quantization->scale()->Get(i); in ParseQuantization()
445 affine_quantization->zero_point->data[i] = in ParseQuantization()
448 affine_quantization->quantized_dimension = in ParseQuantization()
450 quantization->params = reinterpret_cast<void*>(affine_quantization); in ParseQuantization()
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2021-156.md12 const auto* affine_quantization =
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc145 auto* affine_quantization = in GetLegacyQuantization() local
147 if (!affine_quantization || !affine_quantization->scale || in GetLegacyQuantization()
148 !affine_quantization->zero_point || in GetLegacyQuantization()
149 affine_quantization->scale->size != 1 || in GetLegacyQuantization()
150 affine_quantization->zero_point->size != 1) { in GetLegacyQuantization()
155 legacy_quantization.scale = affine_quantization->scale->data[0]; in GetLegacyQuantization()
156 legacy_quantization.zero_point = affine_quantization->zero_point->data[0]; in GetLegacyQuantization()