From 43433fa10921e778137663a08b0dbb418adbcc99 Mon Sep 17 00:00:00 2001
From: Lianmin Zheng <mercy_zheng@sjtu.edu.cn>
Date: Fri, 14 Dec 2018 10:00:46 -0800
Subject: [PATCH] [RELAY] Fix alter_op_layout (#2289)
---
python/tvm/relay/op/nn/_nn.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/python/tvm/relay/op/nn/_nn.py b/python/tvm/relay/op/nn/_nn.py
index f5f76e6af..8180d8b31 100644
--- a/python/tvm/relay/op/nn/_nn.py
+++ b/python/tvm/relay/op/nn/_nn.py
@@ -107,6 +107,12 @@ def schedule_conv2d(attrs, outs, target):
return topi.generic.schedule_depthwise_conv2d_nhwc(outs)
raise ValueError("No compatible schedule")
+
+@reg.register_alter_op_layout("nn.conv2d")
+def alter_op_layout_conv2d(attrs, inputs, tinfos):
+ """Alternate the layout of conv2d"""
+ return None
+
reg.register_pattern("nn.conv2d", OpPattern.OUT_ELEMWISE_FUSABLE)
--
GitLab