prepare_for_propagation_comparison — PyTorch 2.7 documentation (original) (raw)

class torch.ao.quantization.prepare_for_propagation_comparison(model)[source]¶

Add output loggers to node that has numeric_debug_handle

Parameters

model (GraphModule) – original model

Returns

a model with output loggers for all nodes that has numeric_debug_handle_id

Return type