Fix convert_nodes.cc to support TRT7.

PiperOrigin-RevId: 295986213 Change-Id: I6221fca9278c02c2bb657b844980d2b2aef21a44
2020-02-19 09:27:06 -08:00 · 2020-02-19 09:27:06 -08:00 · df36e873c5
commit df36e873c5
parent 0085ae0f60
1 changed files with 3 additions and 0 deletions
--- a/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc
+++ b/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc
@ -660,6 +660,9 @@ size_t TRT_ShapedWeights::size_bytes() const {
      data_type_size = 2;
      break;
    case nvinfer1::DataType::kINT8:
+#if IS_TRT_VERSION_GE(7, 0, 0, 0)
+    case nvinfer1::DataType::kBOOL:
+#endif
      data_type_size = 1;
      break;
  }