From 967f25114e051e1f999c5722fc0d72678f5f7653 Mon Sep 17 00:00:00 2001 From: Trevor Morris Date: Mon, 4 Jan 2021 18:22:02 +0000 Subject: [PATCH] ReshapeAttrs no longer has reverse --- 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 1e6867b83cff..69bb1dccfb62 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>("reverse")[0]), false); auto str_newshape = params->node.GetAttr>("newshape"); std::vector new_shape; const int start_index = TRT_HAS_IMPLICIT_BATCH(params) ? 1 : 0;