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 <[email protected]>
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;