4013b930923dd45ed6a0960505fd04f5a88d7428,hls4ml/model/optimizer/passes/bn_quant.py,QuantizeDenseOutput,transform,#QuantizeDenseOutput#Any#Any#,118

Before Change

        quantized_data = None
        quantized_precision = None
        if node.get_attr("quantize") == 2:
            quantized_data = model.quantize_data(node.weights["weight"].data, 1)
            quantized_precision = "ap_uint<1>"
        elif node.get_attr("quantize") == 3:

After Change

            quantized_precision = hls_model.IntegerPrecisionType(width=2)
        else:
            print("WARNING: Unknown quantizer - {}. Bailing out".format(quantizer.__class__.__name__))
            return False
        quantizer.bits = quantized_precision.width
        quantizer.hls_type = quantized_precision
        quantized_data = quantizer(node.weights["weight"].data)

Instances