diff --git a/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc b/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc index 82bd8d4592f..433564513db 100644 --- 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; }