fix

python/tvm/autotvm/graph_tuner/base_graph_tuner.py

@@ -35,13 +35,6 @@
 from ._base import INVALID_LAYOUT_TIME
 
 
-# Setup topi_op_name -> layout function
-# NOTE: To add more ops, change the following dictionary.
-# OP2LAYOUT = {
-#     "topi_nn_conv2d": topi.nn.conv2d_infer_layout,
-#     "topi_nn_depthwise_conv2d_nchw": topi.nn.depthwise_conv2d_infer_layout,
-# }
-
 def get_infer_layout(task_name):
     if task_name.startswith("conv2d"):
         return topi.nn.conv2d_infer_layout

python/tvm/autotvm/graph_tuner/utils/traverse_graph.py

@@ -140,13 +140,9 @@ def _traverse_expr(node):
                 _expr2graph_impl(node, target_ops, node_dict, node_list)
             return
         elif isinstance(node, TupleGetItem):
-            # TODO(@icemelon9): figure out why we need this?
-            node_entry["op"] = "TupleGetItem"
             in_node_idx = node_dict[node.tuple_value]
             node_entry["inputs"].append([in_node_idx, node.index, 0])
         elif isinstance(node, Tuple):
-            # TODO(@icemelon9): figure out why we need this?
-            node_entry["op"] = "Tuple"
             for tuple_item in node:
                 in_node_idx = node_dict[tuple_item]
                 if isinstance(tuple_item, TupleGetItem):

topi/python/topi/argwhere.py

@@ -16,7 +16,6 @@
 # under the License.
 # pylint: disable=invalid-name, too-many-arguments, too-many-nested-blocks
 """Argwhere operator"""
-import tvm
 from tvm import hybrid
 
 @hybrid.script

vta/scripts/tune_resnet.py

@@ -246,7 +246,7 @@ def tune_tasks(tasks,
     print("Extracting tasks...")
     tasks = extract_from_program(func=relay_prog,
                                  params=params,
-                                 ops=(relay.op.get("nn.conv2d"),)
+                                 ops=(relay.op.get("nn.conv2d"),),
                                  target=target,
                                  target_host=env.target_host)
 

vta/tutorials/autotvm/tune_relay_vta.py

@@ -350,7 +350,7 @@ def tune_and_evaluate(tuning_opt):
     relay_prog, params = compile_network(env, target, network, start_pack, stop_pack)
     tasks = autotvm.task.extract_from_program(func=relay_prog,
                                               params=params,
-                                              ops=(relay.op.get("nn.conv2d"),)
+                                              ops=(relay.op.get("nn.conv2d"),),
                                               target=target,
                                               target_host=env.target_host)