Skip to content

T
tic
Commit 53e84f00 by Steven S. Lyubomirsky Committed by Tianqi Chen
Options

Outdated renaming for flatten in ONNX converter (#2843)

parent 891c4117
Showing with 1 additions and 1 deletions
python/tvm/relay/frontend/onnx.py
......@@ -835,7 +835,7 @@ def _get_convert_map(opset):
# 'InstanceNormalization'
# 'LpNormalization'
'Dropout': AttrCvt('dropout', {'ratio': 'rate'}, ignores=['is_test']),
'Flatten': Renamer('flatten'),
'Flatten': Renamer('batch_flatten'),
'LRN': LRN.get_converter(opset),
# 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