onnx_extended.helper#

make_dynamic_quantize_linear_function_proto#

onnx_extended.helper.make_dynamic_quantize_linear_function_proto(domain: str, opset: int, to: int | None = None) → FunctionProto[source]#

Creates the FunctionProto for function DynamicQuantizeLinear doing a quantization to float 8.

Parameters:

  • domain – local domain name

  • opset – opset to use to define the function

  • to – if None, the function has an attribute, otherwise, it is replaced by the given value

Returns:

FunctionProto

The function takes 1 input and returns 3 outputs like operator DynamicQuantizeLinear. It has one attribute to which specified the quantized type.

make_simple_dynamic_quantize_linear_function_proto#

onnx_extended.helper.make_simple_dynamic_quantize_linear_function_proto(domain: str, opset: int, to: int = 17) → FunctionProto[source]#

Creates the FunctionProto for function SimpleDynamicQuantizeLinear doing a quantization to float 8. A suffix is added to the function name to tell which type is used for the quantization. It does not support nan values.

Parameters:

  • domain – local domain name

  • opset – opset to use to define the function

  • to – type to quantize into, it is hardcoded

Returns:

FunctionProto

The function takes 1 input and returns 3 outputs like operator DynamicQuantizeLinear. It has one attribute to which specified the quantized type.