prepare_for_propagation_comparison

class torch.ao.quantization.prepare_for_propagation_comparison(model)

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

参数

model (GraphModule) – 原始模型

返回值

为所有具有 numeric_debug_handle_id 的节点添加输出记录器的模型

返回类型

GraphModule