prepare_for_propagation_comparison¶
- class torch.ao.quantization.prepare_for_propagation_comparison(model)¶
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
