AlterOpLayout with tvm.target (#463)

* AlterOpLayout with tvm.target

* fix test

nnvm/python/nnvm/compiler/build_module.py

@@ -264,6 +264,7 @@ def build(graph, target=None, shape=None, dtype="float32",
     if _all_var_init:
         init_var = initialize_variables(shape, dtype)
     # Apply optimization
+    with target:
         graph = optimize(graph, shape, dtype, layout)
     # Precompute prune
     if params and cfg.pass_enabled("PrecomputePrune"):