experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Improve comment

Browse Source
This commit is contained in:
Trevor Morris 2019-02-14 11:24:29 -08:00
parent b839268499
commit 95e5de50f0
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc
View File

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