From 5e907a67d58d8b118f3b74797ec4b04b6dad8a4a Mon Sep 17 00:00:00 2001
From: Cavus Mustafa <mustafa.cavus@intel.com>
Date: Thu, 29 Feb 2024 18:44:19 -0800
Subject: [PATCH] Additional op support for TorchFX

---
 .../src/openvino/frontend/pytorch/torchdynamo/op_support.py    | 3 ++-
 src/frontends/pytorch/src/op_table.cpp                         | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/bindings/python/src/openvino/frontend/pytorch/torchdynamo/op_support.py b/src/bindings/python/src/openvino/frontend/pytorch/torchdynamo/op_support.py
index 1790762c9a242d..0b4e69624c4aaa 100644
--- a/src/bindings/python/src/openvino/frontend/pytorch/torchdynamo/op_support.py
+++ b/src/bindings/python/src/openvino/frontend/pytorch/torchdynamo/op_support.py
@@ -111,6 +111,7 @@ def __init__(self, options):
             "torch.ops.aten.expand.default": None,
             "torch.ops.aten.fake_quantize_per_channel_affine_cachemask.default": None,
             "torch.ops.aten.fill.Scalar": None,
+            "torch.ops.aten.fill_.Scalar": None,
             "torch.ops.aten.fill.Tensor": None,
             "torch.ops.aten.fill_.Tensor": None,
             "torch.ops.aten.flip.default": None,
@@ -225,7 +226,6 @@ def __init__(self, options):
             "torch.ops.aten.unbind.int": None,
             "torch.ops.aten.unfold.default": None,
             "torch.ops.aten.unsqueeze.default": None,
-            "torch.ops.aten.upsample_bilinear2d.default": None,
             "torch.ops.aten.upsample_nearest2d.default": None,
             "torch.ops.aten.var.correction": None,
             "torch.ops.aten.var_mean.correction": None,
@@ -233,6 +233,7 @@ def __init__(self, options):
             "torch.ops.aten.where.self": None,
             "torch.ops.aten.zeros_like.default": None,
             "torch.ops.torchvision.deform_conv2d.default": None,
+            "torch.ops.torchvision.roi_align.default": None,
         }
 
         for op in _get_disabled_ops(options):
diff --git a/src/frontends/pytorch/src/op_table.cpp b/src/frontends/pytorch/src/op_table.cpp
index d926a1a8e073c4..3f91f55ae42272 100644
--- a/src/frontends/pytorch/src/op_table.cpp
+++ b/src/frontends/pytorch/src/op_table.cpp
@@ -769,6 +769,7 @@ const std::map<std::string, CreatorFunction> get_supported_ops_fx() {
         {"aten.expand.default", op::translate_expand_fx},
         {"aten.fake_quantize_per_channel_affine_cachemask.default", op::translate_fake_quantize_per_channel_affine_fx},
         {"aten.fill.Scalar", op::translate_fill},
+        {"aten.fill_.Scalar", op::inplace_op<op::translate_fill>},
         {"aten.fill.Tensor", op::translate_fill},
         {"aten.fill_.Tensor", op::inplace_op<op::translate_fill>},
         {"aten.flip.default", op::translate_flip},
@@ -883,7 +884,6 @@ const std::map<std::string, CreatorFunction> get_supported_ops_fx() {
         {"aten.unbind.int", op::translate_unbind_int_fx},
         {"aten.unfold.default", op::translate_unfold},
         {"aten.unsqueeze.default", op::translate_1to1_match_2_inputs<opset10::Unsqueeze>},
-        {"aten.upsample_bilinear2d.default", op::translate_upsample_bilinear2d},
         {"aten.upsample_nearest2d.default", op::translate_upsample_nearest2d},
         {"aten.var.correction", op::translate_var_fx},
         {"aten.var_mean.correction", op::translate_var_mean_fx},