compare_results — PyTorch 2.7 documentation (original) (raw)

class torch.ao.quantization.compare_results(ref_results, actual_results)[source]¶

Given two dict mapping from debug_handle_id (int) to list of tensors return a map from debug_handle_id to NodeAccuracySummary that contains comparison information like SQNR, MSE etc.

Parameters

• ref_results (Dict [_int,_ Tuple [_str,_ object, List _[_torch.Tensor] ] ]) – reference results for each debug_handle_id
• actual_results (Dict [_int,_ Tuple [_str,_ object, List _[_torch.Tensor] ] ]) – actual results for each debug_handle_id

Returns

Dict[int, NodeAccuracySummary]

Return type

dict[int, torch.ao.quantization.pt2e._numeric_debugger.NodeAccuracySummary]