Outdated renaming for flatten in ONNX converter (apache#2843)

wweic · Apr 8, 2019 · 464c0af · 464c0af
1 parent a4d17ce
commit 464c0af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/frontend/onnx.py b/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