Python QuantConv2d.quant_weight_scale Examples

Example #1

 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

Example #2

 def quant_weight_scale(module: QuantConv2d):
     quant_weight_scale = module.quant_weight_scale()
     return DPUv1QuantConv2dHandler.neg_scalar_exponent_from_scale(
         quant_weight_scale)