prepare_for_propagation_comparison

class torch.ao.quantization.prepare_for_propagation_comparison(model)

为具有 numeric_debug_handle 的节点添加输出记录器

参数

model (GraphModule) – 原始模型

返回

一个模型，其中包含具有 numeric_debug_handle_id 的所有节点的输出记录器

返回类型

GraphModule