Commit 53e84f00 by Steven S. Lyubomirsky Committed by Tianqi Chen

Outdated renaming for flatten in ONNX converter (#2843)

parent 891c4117
...@@ -835,7 +835,7 @@ def _get_convert_map(opset): ...@@ -835,7 +835,7 @@ def _get_convert_map(opset):
# 'InstanceNormalization' # 'InstanceNormalization'
# 'LpNormalization' # 'LpNormalization'
'Dropout': AttrCvt('dropout', {'ratio': 'rate'}, ignores=['is_test']), 'Dropout': AttrCvt('dropout', {'ratio': 'rate'}, ignores=['is_test']),
'Flatten': Renamer('flatten'), 'Flatten': Renamer('batch_flatten'),
'LRN': LRN.get_converter(opset), 'LRN': LRN.get_converter(opset),
# defs/reduction # defs/reduction
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment