failed to legalize operation 'torch.operator' that was explicitly marked illegal : onnx.NonMaxSuppression

[pdhirajkumarprasad]

For the given IR, getting error ' failed to legalize operation 'torch.operator' that was explicitly marked illegal : onnx.NonMaxSuppression'

module {
  func.func @main_graph( %arg1: !torch.vtensor<[1,?,4],f32>, %arg2: !torch.vtensor<[1,1,?],f32>) -> (!torch.vtensor<[?,3],si64> ) attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 21 : si64, torch.onnx_meta.producer_name = "pytorch", torch.onnx_meta.producer_version = "2.1.0"} {
    %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense_resource<__329> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> 
    %2 = torch.operator "onnx.Constant"() {torch.onnx.value = dense_resource<__330> : tensor<1xf32>} : () -> !torch.vtensor<[1],f32> 
    %3 = torch.operator "onnx.NonMaxSuppression"(%arg1, %arg2, %1, %2) : (!torch.vtensor<[1,?,4],f32>, !torch.vtensor<[1,1,?],f32>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],f32>) -> !torch.vtensor<[?,3],si64> 
    return %3: !torch.vtensor<[?,3],si64>
  }
}

{-#
  dialect_resources: {
    builtin: {
      __329: "0x08000000FFFFFFFFFFFFFF7F",
      __330: "0x080000003333333F"
    }
  }
#-}

[pdhirajkumarprasad]

model impacted : 24

fasterrcnn_mobilenet_v3_large_320_fpn_Opset17_torchvision
maskrcnn_resnet50_fpn_v2_Opset16_torchvision
retinanet_resnet50_fpn_v2_Opset18_torchvision
retinanet_resnet50_fpn_v2_Opset17_torchvision
fcos_resnet50_fpn_Opset17_torchvision
fasterrcnn_mobilenet_v3_large_fpn_Opset16_torchvision
fasterrcnn_mobilenet_v3_large_fpn_Opset18_torchvision
fasterrcnn_mobilenet_v3_large_320_fpn_Opset18_torchvision
fasterrcnn_mobilenet_v3_large_320_fpn_Opset16_torchvision
MaskRCNN-12
MaskRCNN-12-qdq
MaskRCNN-10
maskrcnn_resnet50_fpn_v2_Opset17_torchvision
fasterrcnn_resnet50_fpn_v2_Opset17_torchvision
retinanet_resnet50_fpn_v2_Opset16_torchvision
FasterRCNN-12
fasterrcnn_resnet50_fpn_v2_Opset18_torchvision
FasterRCNN-12-qdq
FasterRCNN-10
fcos_resnet50_fpn_Opset18_torchvision
fcos_resnet50_fpn_Opset16_torchvision
fasterrcnn_mobilenet_v3_large_fpn_Opset17_torchvision
maskrcnn_resnet50_fpn_v2_Opset18_torchvision
fasterrcnn_resnet50_fpn_v2_Opset16_torchvision