From 63d94077c1ab6eff43e3794007178929d3fde957 Mon Sep 17 00:00:00 2001 From: "Steven S. Lyubomirsky" Date: Sat, 30 Mar 2019 11:19:50 -0700 Subject: [PATCH] Outdated renaming for flatten in ONNX converter (#2843) --- 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 f0f0356e1cb0..45471e1eb892 100644 --- 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