Searched refs:channel_dim_size (Results 1 – 2 of 2) sorted by relevance
86 const int32_t channel_dim_size = dimension[channel_dim_index]; in SymmetricPerChannelQuantization() local87 std::vector<float> min_vals(channel_dim_size); in SymmetricPerChannelQuantization()88 std::vector<float> max_vals(channel_dim_size); in SymmetricPerChannelQuantization()89 std::vector<bool> has_min_max_value(channel_dim_size, false); in SymmetricPerChannelQuantization()118 std::vector<float> scale_invs(channel_dim_size); in SymmetricPerChannelQuantization()120 for (size_t channel_idx = 0; channel_idx < channel_dim_size; channel_idx++) { in SymmetricPerChannelQuantization()
100 const int32_t channel_dim_size = tensor->shape[channel_dim_index]; in SymmetricPerChannelQuantizeTensor() local107 std::vector<float> scales(channel_dim_size); in SymmetricPerChannelQuantizeTensor()148 int32_t channel_dim_size = tensor->shape[0]; in SymmetricPerChannelBiasQuantize() local157 if (weight_scales.size() != channel_dim_size) { in SymmetricPerChannelBiasQuantize()164 std::vector<float> scales(channel_dim_size); in SymmetricPerChannelBiasQuantize()165 for (size_t i = 0; i < channel_dim_size; i++) { in SymmetricPerChannelBiasQuantize()177 for (int32_t channel_idx = 0; channel_idx < channel_dim_size; channel_idx++) { in SymmetricPerChannelBiasQuantize()