From 3edbd43052d617262a2fc1d7f091ef4e33df5ac8 Mon Sep 17 00:00:00 2001 From: Haichen Shen Date: Tue, 10 Sep 2019 18:55:33 +0000 Subject: [PATCH] [Opt] Use opaque for where op --- src/relay/op/tensor/transform.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/relay/op/tensor/transform.cc b/src/relay/op/tensor/transform.cc index 203a0411d3c4..6cd79b102010 100644 --- a/src/relay/op/tensor/transform.cc +++ b/src/relay/op/tensor/transform.cc @@ -1601,7 +1601,7 @@ Examples:: .set_support_level(4) .add_type_rel("Where", WhereRel) .set_attr("FTVMCompute", WhereCompute) -.set_attr("TOpPattern", kBroadcast); +.set_attr("TOpPattern", kOpaque); // Squeeze