diff --git a/python/tvm/autotvm/task/nnvm_integration.py b/python/tvm/autotvm/task/nnvm_integration.py index 8a8d33dd5ac80..a586dfbe2a2f0 100644 --- 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: