Skip to content

Commit 7c37ccb

Browse files
authored
Merge pull request #903 from jignparm/master
Some ops for opset 12.
2 parents 40dd461 + b78a433 commit 7c37ccb

File tree

3 files changed

+15
-0
lines changed

3 files changed

+15
-0
lines changed

tf2onnx/onnx_opset/math.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -137,6 +137,9 @@ def version_1(cls, ctx, node, **kwargs):
137137
ctx.remove_node(node.name)
138138
make_min_or_max_op(ctx, node.type, node.input, node.output, shapes, dtypes)
139139

140+
@classmethod
141+
def version_12(cls, ctx, node, **kwargs):
142+
pass # support all numeric types and broadcasting
140143

141144
@tf_op("ClipByValue")
142145
class ClipByValueOp:
@@ -178,6 +181,9 @@ def version_8(cls, ctx, node, **kwargs):
178181
ctx.set_dtype(new_node.output[0], dtypes[0])
179182
ctx.set_shape(new_node.output[0], shapes[0])
180183

184+
@classmethod
185+
def version_12(cls, ctx, node, **kwargs):
186+
node.name = 'Clip' # clip supports all types now
181187

182188
@tf_op("Softmax")
183189
class Softmax:

tf2onnx/onnx_opset/reduction.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,11 @@ def version_11(cls, ctx, node, **kwargs):
8989
# Opset 11 supports negative axis, but core logic same
9090
cls.version_1(ctx, node, **kwargs)
9191

92+
@classmethod
93+
def version_12(cls, ctx, node, **kwargs):
94+
# Opset 12 adds extra attribute 'select_last_index'
95+
# No changes needed
96+
cls.version_1(ctx, node, **kwargs)
9297

9398
@tf_op(["All", "Any"])
9499
class AllAny:

tf2onnx/onnx_opset/tensor.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -100,6 +100,10 @@ def version_7(cls, ctx, node, **kwargs):
100100
def version_10(cls, ctx, node, **kwargs):
101101
pass
102102

103+
@classmethod
104+
def version_12(cls, ctx, node, **kwargs):
105+
pass
106+
103107

104108
@tf_op("Identity")
105109
class Identity:

0 commit comments

Comments
 (0)