dmx.compressor.numerical.cast.NumericalCastMixin

class dmx.compressor.numerical.cast.NumericalCastMixin(*args, **kwargs)

Mixin for modules with boundary casting

__init__(*args, **kwargs) → None

Methods

__init__(*args, **kwargs)
check_format_dim_consistency()
check_input_format_dim_consistency()
check_residual_format_dim_consistency()
check_weight_format_dim_consistency()
infer_ch_axis()
init_casts()
init_smoothquant([migration_strength, ...])

Attributes

accum_format
bias_format
input_formats
input_precision
multiplier_format
output_formats
residual_format
weight_format
weight_precision
weight_scale
weight_storage_format
weight_storage_precision
weight_storage_scale
weight_storage_zero_point
weight_zero_point