This is an automated email from the ASF dual-hosted git repository. comaniac pushed a commit to branch main in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push: new 7163b5c ReshapeAttrs no longer has reverse (#7205) 7163b5c is described below commit 7163b5c02fd25326e7c68ccc3b41d30f4a912952 Author: Trevor Morris <trevm...@amazon.com> AuthorDate: Mon Jan 4 14:01:53 2021 -0800 ReshapeAttrs no longer has reverse (#7205) --- src/runtime/contrib/tensorrt/tensorrt_ops.cc | 1 - 1 file changed, 1 deletion(-) diff --git a/src/runtime/contrib/tensorrt/tensorrt_ops.cc b/src/runtime/contrib/tensorrt/tensorrt_ops.cc index 1e6867b..69bb1dc 100644 --- a/src/runtime/contrib/tensorrt/tensorrt_ops.cc +++ b/src/runtime/contrib/tensorrt/tensorrt_ops.cc @@ -921,7 +921,6 @@ class ReshapeOpConverter : public TensorRTOpConverter { void Convert(TensorRTOpConverterParams* params) const { auto input = params->inputs.at(0).tensor; - ICHECK_EQ(std::stoi(params->node.GetAttr<std::vector<std::string>>("reverse")[0]), false); auto str_newshape = params->node.GetAttr<std::vector<std::string>>("newshape"); std::vector<int> new_shape; const int start_index = TRT_HAS_IMPLICIT_BATCH(params) ? 1 : 0;