Skip to content

Commit 75e5c03

Browse files
committed
rename custom folder to custom_opsets
1 parent 56670ab commit 75e5c03

File tree

3 files changed

+2
-2
lines changed

3 files changed

+2
-2
lines changed
File renamed without changes.
File renamed without changes.

tf2onnx/tfonnx.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
from tensorflow.tools.graph_transforms import TransformGraph
2222

2323
import tf2onnx
24-
from tf2onnx import constants, custom, schemas, utils
24+
from tf2onnx import constants, custom_opsets, schemas, utils
2525
from tf2onnx.function import * # pylint: disable=wildcard-import
2626
from tf2onnx.graph import Graph
2727
from tf2onnx.graph_matcher import OpTypePattern, GraphMatcher
@@ -2336,7 +2336,7 @@ def tensorflow_onnx_mapping(g, continue_on_error, custom_op_handlers):
23362336
for extra_opset in g.extra_opset:
23372337
if extra_opset.domain == constants.MICROSOFT_DOMAIN:
23382338
# microsoft domain
2339-
for target_opset, op_map in custom.ms.OPSETS:
2339+
for target_opset, op_map in custom_opsets.ms.OPSETS:
23402340
if target_opset <= extra_opset.version:
23412341
ops_mapping.update(op_map)
23422342
else:

0 commit comments

Comments
 (0)