From 1b6932a92021d56862cd6159856f995b5c90eb5b Mon Sep 17 00:00:00 2001 From: "Steven S. Lyubomirsky" Date: Mon, 18 Mar 2019 14:02:18 -0700 Subject: [PATCH] Outdated renaming for flatten in ONNX converter --- python/tvm/relay/frontend/onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py index d322da31fc19..7a55eb705b53 100644 --- a/python/tvm/relay/frontend/onnx.py +++ b/python/tvm/relay/frontend/onnx.py @@ -807,7 +807,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