update fuzzy test list

openvino-dev-samples · Sep 3, 2024 · 37f2b79 · 37f2b79
1 parent 8fa5f26
commit 37f2b79
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/frontends/paddle/src/op/roll.cpp b/src/frontends/paddle/src/op/roll.cpp
@@ -13,7 +13,9 @@ NamedOutputs roll(const NodeContext& node) {
     auto input_node = node.get_input("X");
     Output<Node> shifts_node;
     if (node.has_input("ShiftsTensor")) {
-        auto shifts_node = node.get_input("ShiftsTensor");
+        auto shifts = node.get_input("ShiftsTensor");
+        auto shifts_var_node = std::make_shared<default_opset::Convert>(shifts, element::i64);
+        shifts_node = std::make_shared<default_opset::Squeeze>(shifts_var_node);
     } else {
         const auto shifts = node.get_attribute<std::vector<int64_t>>("shifts");
         shifts_node = default_opset::Constant::create(element::i64, {shifts.size()}, shifts);

diff --git a/src/frontends/paddle/tests/op_fuzzy.cpp b/src/frontends/paddle/tests/op_fuzzy.cpp
@@ -513,6 +513,7 @@ static const std::vector<std::string> models{
     std::string("roll_test_1/roll_test_1.pdmodel"),
     std::string("roll_test_2/roll_test_2.pdmodel"),
     std::string("roll_test_3/roll_test_3.pdmodel"),
+    std::string("roll_test_4/roll_test_4.pdmodel"),
     std::string("round/round.pdmodel"),
     std::string("scale_bias_after_float32/scale_bias_after_float32.pdmodel"),
     std::string("scale_bias_after_int32/scale_bias_after_int32.pdmodel"),