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