Searched refs:quantized_dimension (Results 1 – 8 of 8) sorted by relevance
339 if (src_quantization->quantized_dimension() < 0 || in ParseQuantization()340 src_quantization->quantized_dimension() >= num_scales) { in ParseQuantization()343 src_quantization->quantized_dimension()); in ParseQuantization()346 affine_quantization->quantized_dimension = in ParseQuantization()347 src_quantization->quantized_dimension(); in ParseQuantization()
38 int quantized_dimension, in AddQuantizationParams() argument48 tensor->quantization->quantized_dimension = quantized_dimension; in AddQuantizationParams()
188 filter_params->quantized_dimension = 0; in TEST_F()297 filter_params->quantized_dimension = 0; in TEST_F()
204 const int channel_index = params->quantized_dimension; in PerChannelSymmetricQuantizeAndPopulate()460 affine_quantization->quantized_dimension = 0; in QuantizeTensor()
67 filter->dims->data[affine_quantization->quantized_dimension]); in PopulateConvolutionQuantizationParams()
251 int32_t quantized_dimension; member
80 quantized_dimension:int;
2491 int32_t quantized_dimension;2493 : quantized_dimension(0) {2530 int32_t quantized_dimension() const {2579 void add_quantized_dimension(int32_t quantized_dimension) {2580 … fbb_.AddElement<int32_t>(QuantizationParameters::VT_QUANTIZED_DIMENSION, quantized_dimension, 0);2602 int32_t quantized_dimension = 0) {2604 builder_.add_quantized_dimension(quantized_dimension);2622 int32_t quantized_dimension = 0) {2631 quantized_dimension);8788 { auto _e = quantized_dimension(); _o->quantized_dimension = _e; };[all …]