Fix ng python ShuffleChannels group attribute (#5678)

This commit is contained in:
Katarzyna Mitrus authored and GitHub committed 2021-05-20 12:31:19 +02:00
1 parent 13d5a53aa4
commit 82b84c0b61
3 files changed
+3 -7

No files matched your search

+2 -2
View File
@@ -555,7 +555,7 @@ def shape_of(data: NodeInput, output_type: str = "i64", name: Optional[str] = No
@nameable_op
def shuffle_channels(data: Node, axis: int, groups: int, name: Optional[str] = None) -> Node:
def shuffle_channels(data: Node, axis: int, group: int, name: Optional[str] = None) -> Node:
"""Perform permutation on data in the channel dimension of the input tensor.
@param data: The node with input tensor.
@@ -603,7 +603,7 @@ def shuffle_channels(data: Node, axis: int, groups: int, name: Optional[str] = N
@endcode
"""
return _get_node_factory_opset3().create(
"ShuffleChannels", [as_node(data)], {"axis": axis, "groups": groups}
"ShuffleChannels", [as_node(data)], {"axis": axis, "group": group}
)
-2
View File
@@ -53,8 +53,6 @@ xfail_issue_35925 = xfail_test(reason="Assertion error - reduction ops results m
xfail_issue_35927 = xfail_test(reason="RuntimeError: B has zero dimension that is not allowable")
xfail_issue_36480 = xfail_test(reason="RuntimeError: [NOT_FOUND] Unsupported property dummy_option "
"by CPU plugin")
xfail_issue_36485 = xfail_test(reason="RuntimeError: Check 'm_group >= 1' failed at "
"/openvino/ngraph/core/src/op/shuffle_channels.cpp:77:")
xfail_issue_36486 = xfail_test(reason="RuntimeError: HardSigmoid operation should be converted "
"to HardSigmoid_IE")
xfail_issue_36487 = xfail_test(reason="Assertion error - mvn operator computation mismatch")
@@ -6,8 +6,7 @@ import pytest
import ngraph as ng
from tests.runtime import get_runtime
from tests import (xfail_issue_36485,
xfail_issue_36486,
from tests import (xfail_issue_36486,
xfail_issue_36487,
xfail_issue_44976)
@@ -276,7 +275,6 @@ def test_squared_difference_operator():
assert np.allclose(result, expected)
@xfail_issue_36485
def test_shuffle_channels_operator():
runtime = get_runtime()