From 95e5de50f05a10422a4b84d2f647e871aaffafc2 Mon Sep 17 00:00:00 2001 From: Trevor Morris Date: Thu, 14 Feb 2019 11:24:29 -0800 Subject: [PATCH] Improve comment --- tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc b/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc index 4cc72e50b24..d8140949191 100644 --- a/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc +++ b/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc @@ -3773,7 +3773,8 @@ tensorflow::Status ConvertGraphDefToEngine( builder->setFp16Mode(true); } else if (precision_mode == TrtPrecisionMode::INT8) { // Setting FP16 mode as well allows TRT to also consider FP16 kernels and - // use them in situations where they are faster than INT8. + // use them in situations where they are faster than INT8 or where INT8 is + // not supported for a given layer. builder->setFp16Mode(true); builder->setInt8Mode(true); if (use_calibration) {