diff --git a/src/tensorrt_utils.cc b/src/tensorrt_utils.cc index 5a58b09..e959488 100644 --- a/src/tensorrt_utils.cc +++ b/src/tensorrt_utils.cc @@ -522,7 +522,7 @@ TensorFormatToString(const nvinfer1::TensorFormat& io_format) case nvinfer1::TensorFormat::kDLA_HWC4: return "DLA_HWC4"; default: - return "INVALID"; + return "UNKNOWN"; } }