mbrookhart commented on a change in pull request #6906:
URL: https://github.com/apache/incubator-tvm/pull/6906#discussion_r522517222
##########
File path: src/relay/op/tensor/transform.cc
##########
@@ -1076,7 +1076,7 @@ Examples::
.set_support_level(3)
.add_type_rel("Take", TakeRel)
.set_attr<FTVMCompute>("FTVMCompute", TakeCompute)
- .set_attr<TOpPattern>("TOpPattern", kInjective);
+ .set_attr<TOpPattern>("TOpPattern", kOpaque);
Review comment:
I think the other option is to edit the relay take function that creates
the op. We could remove the normalization that causes this problem from the
take kernel in topi, and do in it relay with select/shape_of, but that might
end up causing some performance degradation, it's hard to predict.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
