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) {