Ejemplos de QuantWeightBiasInputOutputLayer.quant_bias_scale en Python

Ejemplo n.º 1

Archivo: handler.py Proyecto: 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

Ejemplo n.º 2

Archivo: handler.py Proyecto: 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

Ejemplo n.º 3

Archivo: test_wbiol.py Proyecto: uslumt/brevitas

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

Ejemplo n.º 4

Archivo: parameter.py Proyecto: fpjentzsch/brevitas

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