Python QuantWeightBiasInputOutputLayer.quant_bias_scaleの例

コード例 #1

ファイル: handler.py プロジェクト: uslumt/brevitas

 def quant_bias_scale(module: QuantWBIOL):
     if module.bias is not None:
         scale = module.quant_bias_scale()
         return DPUQuantLayerHandler.neg_scalar_exponent_from_scale(scale)
     else:
         return None

コード例 #2

ファイル: handler.py プロジェクト: fpjentzsch/brevitas

 def quant_bias_scale(cls, module: QuantWBIOL):
     if module.bias is not None:
         scale = module.quant_bias_scale()
         return cls.validate_neg_scalar_int_exponent(scale)
     else:
         return None

コード例 #3

ファイル: test_wbiol.py プロジェクト: uslumt/brevitas

def test_default_wbiol_quant_bias_scale(default_wbiol_layer: QuantWBIOL):
    assert default_wbiol_layer.quant_bias_scale() is None

コード例 #4

ファイル: parameter.py プロジェクト: fpjentzsch/brevitas

 def maybe_quant_bias_scale(module: QuantWBIOL):
     if module.is_bias_quant_enabled:
         return module.quant_bias_scale()
     else:
         return None