dmx.compressor.fx.transform.qDq_transform

dmx.compressor.fx.transform.qDq_transform(root: GraphModule)

A function that transforms the model by substituting CastTos with Q/dQ ops

Parameters:

root (torch.nn.Module) – model/module to transform

Returns:

transformed model