rebase bug fix (apache#26)

jroesch · Dec 1, 2018 · 826be25 · 826be25
1 parent c632056
commit 826be25
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/autotvm/task/nnvm_integration.py b/python/tvm/autotvm/task/nnvm_integration.py
@@ -89,11 +89,11 @@ def __init__(self, wanted_symbols):
         self.func_to_reflection = {
             topi.nn.conv2d:                lambda x: setattr(topi.nn, 'conv2d', x),
             topi.nn.depthwise_conv2d_nchw: lambda x: setattr(topi.nn, 'depthwise_conv2d_nchw', x),
+            topi.nn.group_conv2d_nchw:     lambda x: setattr(topi.nn, 'group_conv2d_nchw', x),
             topi.nn.conv2d_transpose_nchw: lambda x: setattr(topi.nn, 'conv2d_transpose_nchw', x),
             topi.nn.dense:                 lambda x: setattr(topi.nn, 'dense', x),
         }
 
-
         self.wanted_topi_funcs = []
         for sym_name in wanted_symbols:
             if sym_name in self.symbol2topi: