Skip to content

Commit be13d6b

Browse files
committed
Merge branch 'ellipse_mask' of https://github.com/jiafatom/tensorflow-onnx into ellipse_mask
2 parents a785beb + f1696b7 commit be13d6b

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

tests/test_backend.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1217,7 +1217,7 @@ def test_strided_slice6(self):
12171217
x_ = x[2, :]
12181218
_ = tf.identity(x_, name=_TFOUTPUT)
12191219
self._run_test_case([_OUTPUT], {_INPUT: x_val})
1220-
1220+
12211221
@skip_caffe2_backend("multiple dims not supported")
12221222
def test_strided_slice7(self):
12231223
x_val = np.arange(5 * 6).astype("float32").reshape(5, 6)
@@ -1239,6 +1239,12 @@ def test_strided_slice7(self):
12391239
_ = tf.identity(x_, name=_TFOUTPUT)
12401240
self._run_test_case([_OUTPUT], {_INPUT: x_val})
12411241

1242+
tf.reset_default_graph()
1243+
x = tf.placeholder(tf.float32, x_val.shape, name=_TFINPUT)
1244+
x_ = tf.strided_slice(x, [0, 1], [3, 4], [1, 1], ellipsis_mask=2)
1245+
_ = tf.identity(x_, name=_TFOUTPUT)
1246+
self._run_test_case([_OUTPUT], {_INPUT: x_val})
1247+
12421248
@skip_caffe2_backend("fails with schema error")
12431249
@check_opset_min_version(7, "batchnorm")
12441250
def test_batchnorm(self):

0 commit comments

Comments
 (0)