Skip to content

Commit 763c0c1

Browse files
committed
[Test] add identity padding testcase to test_remove_identity_ops
1 parent 279f9c3 commit 763c0c1

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

tests/transformation/test_remove_identity_ops.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,8 @@
4141

4242

4343
def insert_identity_op(model, op, as_first_node, approx):
44+
kwargs = {}
45+
inp_ndims = 4 if as_first_node else 2
4446
if approx:
4547
zero_val = 0.000001
4648
one_val = 0.999999
@@ -53,6 +55,9 @@ def insert_identity_op(model, op, as_first_node, approx):
5355
val = np.asarray([one_val], dtype=np.float32)
5456
elif op in ["Identity"]:
5557
val = None
58+
elif op == "Pad":
59+
# opset 11 and above: padding specified as input and not attribute
60+
val = np.asarray([0] * 2 * inp_ndims, dtype=np.int64)
5661
else:
5762
return
5863

@@ -62,7 +67,7 @@ def insert_identity_op(model, op, as_first_node, approx):
6267
else:
6368
model.set_initializer("value", val)
6469
inplist = ["inp" if as_first_node else "div_out", "value"]
65-
identity_node = helper.make_node(op, inplist, ["ident_out"])
70+
identity_node = helper.make_node(op, inplist, ["ident_out"], **kwargs)
6671
if as_first_node:
6772
graph.node.insert(0, identity_node)
6873
graph.node[1].input[0] = "ident_out"
@@ -74,7 +79,7 @@ def insert_identity_op(model, op, as_first_node, approx):
7479

7580

7681
# identity operations to be inserted
77-
@pytest.mark.parametrize("op", ["Add", "Sub", "Mul", "Div", "Identity"])
82+
@pytest.mark.parametrize("op", ["Add", "Sub", "Mul", "Div", "Identity", "Pad"])
7883
@pytest.mark.parametrize("approx", [False, True])
7984
@pytest.mark.parametrize("as_first_node", [False, True])
8085
@pytest.mark.parametrize("fork_before_id", [False, True])

0 commit comments

Comments
 (0)