dmx.compressor.fx.transform.make_compiler_graph

dmx.compressor.fx.transform.make_compiler_graph(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