prepare_for_propagation_comparison#
- 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