prepare_for_propagation_comparison¶
- class torch.ao.quantization.prepare_for_propagation_comparison(model)[源代码]¶
向具有 numeric_debug_handle 的节点添加输出记录器
- 参数
model (GraphModule) – 原始模型
- 返回
返回一个模型,该模型包含所有具有 numeric_debug_handle_id 的节点的输出记录器
- 返回类型
向具有 numeric_debug_handle 的节点添加输出记录器
model (GraphModule) – 原始模型
返回一个模型,该模型包含所有具有 numeric_debug_handle_id 的节点的输出记录器