Python QuantWeightBiasInputOutputLayer.int_weight示例

示例#1

 def int_weight(module: QuantWBIOL):
     int_weight = module.int_weight(float_datatype=False).detach()
     return int_weight.type(torch.int8)

示例#2

文件： parameter.py 项目： Xilinx/brevitas

 def int_weight(module: QuantWBIOL):
     int_weight = module.int_weight(float_datatype=False).detach()
     if module.is_quant_weight_signed:
         return int_weight.type(torch.int8)
     else:
         return int_weight.type(torch.uint8)

示例#3

 def int_weight(module: QuantWBIOL):
     return torch.t(module.int_weight(float_datatype=True)).detach()

示例#4

文件： handler.py 项目： uslumt/brevitas

 def int_weight(module: QuantWBIOL):
     return module.int_weight(float_datatype=False).detach()