DataType¶
-
tensorrt.
DataType
¶ Represents data types.
Members:
FLOAT : Represents a 32-bit floating point number.
INT8 : Represents an 8-bit integer.
INT32 : Represents a 32-bit integer.
HALF : Represents a 16-bit floating point number.
TensorRT also exposes some DataType
aliases that can be used across the library.
tensorrt.float32
corresponds toDataType.FLOAT
tensorrt.float16
corresponds toDataType.HALF
tensorrt.int32
corresponds toDataType.INT32
tensorrt.int8
corresponds toDataType.INT8
The size in bytes of a data type can be retrieved with tensorrt.DataType.itemsize
. For example, tensorrt.float32.itemsize
would return 4
.