def quant_weight_scale(module: QuantConv2d): quant_weight_scale = module.quant_weight_scale().type( torch.FloatTensor).detach() if len(quant_weight_scale.shape) == 4: quant_weight_scale = quant_weight_scale.view(1, -1, 1, 1) return quant_weight_scale
def quant_weight_scale(module: QuantConv2d): quant_weight_scale = module.quant_weight_scale() return DPUv1QuantConv2dHandler.neg_scalar_exponent_from_scale( quant_weight_scale)