Skip to content

Commit f1696b7

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

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tf2onnx/tfonnx.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -938,6 +938,8 @@ def stridedslice_op(ctx, node, name, args):
938938
max_size = np.iinfo(begin.dtype).max
939939
end_mask = node.get_attr("end_mask")
940940
end_mask = end_mask.i if end_mask is not None else 0
941+
ellipsis_mask = node.get_attr("ellipsis_mask")
942+
ellipsis_mask = ellipsis_mask.i if ellipsis_mask is not None else 0
941943
begin_mask = node.get_attr("begin_mask")
942944
begin_mask = begin_mask.i if begin_mask is not None else 0
943945
shrink_axis_mask = node.get_attr("shrink_axis_mask")

0 commit comments

Comments
 (0)