| commit | ccf0faf8fb22c1deb67b5409684f84a6300e3f03 | [log] [tgz] |
|---|---|---|
| author | Matt Conley <mconley@nvidia.com> | Thu Jun 24 13:53:35 2021 -0700 |
| committer | Matt Conley <mconley@nvidia.com> | Thu Jun 24 13:53:35 2021 -0700 |
| tree | 4c69a369e29ac8b2ff3312b4be4407643137a626 | |
| parent | cd336ebc4ca0c202476bfa54cc1b7e9e4dd5f105 [diff] |
Correct SimpleITensor Assertion
diff --git a/tensorflow/compiler/tf2tensorrt/utils/trt_tensor_proxy.h b/tensorflow/compiler/tf2tensorrt/utils/trt_tensor_proxy.h index 3eacca2..789c518 100644 --- a/tensorflow/compiler/tf2tensorrt/utils/trt_tensor_proxy.h +++ b/tensorflow/compiler/tf2tensorrt/utils/trt_tensor_proxy.h
@@ -163,7 +163,7 @@ SimpleITensor* simple_tensor() const { assert(simple_tensor_ != nullptr); - assert(ttype_ == TensorType::kTRT); + assert(ttype_ == TensorType::kSIMPLE); return simple_tensor_.get(); }