Searched refs:channel_dim_size (Results 1 – 2 of 2) sorted by relevance
158 const int32_t channel_dim_size = dimension[channel_dim_index]; in FillPerChannelMinMax() local160 quantization_params->min = std::vector<float>(channel_dim_size); in FillPerChannelMinMax()161 quantization_params->max = std::vector<float>(channel_dim_size); in FillPerChannelMinMax()162 std::vector<bool> has_min_max_value(channel_dim_size, false); in FillPerChannelMinMax()248 int channel_dim_size = quant_params->min.size(); in AdjustWeightsForBiasScale() local249 if (channel_dim_size == 0) { in AdjustWeightsForBiasScale()257 std::vector<float> weight_scales(channel_dim_size); in AdjustWeightsForBiasScale()262 if (channel_dim_size > 1) { in AdjustWeightsForBiasScale()263 for (int i = 0; i < channel_dim_size; ++i) { in AdjustWeightsForBiasScale()273 } else if (channel_dim_size == 1) { in AdjustWeightsForBiasScale()[all …]
151 int32_t channel_dim_size = bias_tensor->shape[0]; in QuantizeBias() local170 if (weight_scales.size() != channel_dim_size) { in QuantizeBias()179 weight_scales.data(), channel_dim_size, error_reporter); in QuantizeBias()183 weight_scales.data(), channel_dim_size, error_reporter); in QuantizeBias()1304 int32_t channel_dim_size = bias_tensor->shape[0]; in EnsureBiasScaleCompatibility() local1371 if (weight_tensor->quantization->max.size() != channel_dim_size) { in EnsureBiasScaleCompatibility()1376 weight_tensor->quantization->max.size(), channel_dim_size); in EnsureBiasScaleCompatibility()1379 if (weight_tensor->quantization->min.size() != channel_dim_size) { in EnsureBiasScaleCompatibility()