TensorRT 7.2 supports operators up to Opset 13. Latest information of ONNX operators can be found here
TensorRT supports the following ONNX data types: DOUBLE, FLOAT32, FLOAT16, INT8, and BOOL
Note: There is limited support for INT32, INT64, and DOUBLE types. TensorRT will attempt to cast down INT64 to INT32 and DOUBLE down to FLOAT where possible. If not possible, TensorRT will throw an error. See the TensorRT layer support matrix for more information on data type support.
Operator | Supported? | Restrictions |
---|---|---|
Abs | Y | |
Acos | Y | |
Acosh | Y | |
Add | Y | |
And | Y | |
ArgMax | Y | |
ArgMin | Y | |
Asin | Y | |
Asinh | Y | |
Atan | Y | |
Atanh | Y | |
AveragePool | Y | 2D or 3D Pooling only |
BatchNormalization | Y | |
BitShift | N | |
Cast | Y | Only supported for TensorRT types |
Ceil | Y | |
Celu | Y | |
Clip | Y | min and max clip values must be initializers |
Compress | N | |
Concat | Y | |
ConcatFromSequence | N | |
Constant | Y | |
ConstantOfShape | Y | |
Conv | Y | 2D or 3D convolutions only |
ConvInteger | N | |
ConvTranspose | Y | 2D or 3D deconvolutions only. Weights W must be an initializer |
Cos | Y | |
Cosh | Y | |
CumSum | Y | axis must be an initializer |
DepthToSpace | Y | |
DequantizeLinear | Y | x_scale and x_zero_point must be initializers |
Det | N | |
Div | Y | |
Dropout | N | |
DynamicQuantizeLinear | N | |
Einsum | N | |
Elu | Y | |
Equal | Y | |
Erf | Y | |
Exp | Y | |
Expand | Y | |
EyeLike | Y | |
Flatten | Y | |
Floor | Y | |
Gather | Y | |
GatherElements | Y | Only positive indices (>=0) are supported |
GatherND | N | |
Gemm | Y | |
GlobalAveragePool | Y | |
GlobalLpPool | Y | |
GlobalMaxPool | Y | |
Greater | Y | |
GreaterOrEqual | Y | |
GRU | Y | |
HardSigmoid | Y | |
Hardmax | N | |
Identity | Y | |
If | N | |
ImageScaler | Y | |
InstanceNormalization | Y | Scales scale and biases B must be initializers |
IsInf | N | |
IsNaN | N | |
LeakyRelu | Y | |
Less | Y | |
LessOrEqual | Y | |
Log | Y | |
LogSoftmax | Y | |
Loop | Y | |
LRN | Y | |
LSTM | Y | |
LpNormalization | Y | |
LpPool | Y | |
MatMul | Y | |
MatMulInteger | N | |
Max | Y | |
MaxPool | Y | |
MaxRoiPool | N | |
MaxUnpool | N | |
Mean | Y | |
MeanVarianceNormalization | N | |
Min | Y | |
Mod | N | |
Mul | Y | |
Multinomial | N | |
Neg | Y | |
NegativeLogLikelihoodLoss | N | |
NonMaxSuppression | N | |
NonZero | N | |
Not | Y | |
OneHot | N | |
Or | Y | |
Pad | Y | Zero-padding on last 2 dimensions only |
ParametricSoftplus | Y | |
Pow | Y | |
PRelu | Y | |
QLinearConv | N | |
QLinearMatMul | N | |
QuantizeLinear | Y | Scales y_scale and zero-point y_zero_point must be initializers |
RandomNormal | N | |
RandomNormalLike | N | |
RandomUniform | Y | |
RandomUniformLike | Y | |
Range | Y | Float inputs are only supported if start , limit , and delta inputs are initializers |
Reciprocal | N | |
ReduceL1 | Y | |
ReduceL2 | Y | |
ReduceLogSum | Y | |
ReduceLogSumExp | Y | |
ReduceMax | Y | |
ReduceMean | Y | |
ReduceMin | Y | |
ReduceProd | Y | |
ReduceSum | Y | |
ReduceSumSquare | Y | |
Relu | Y | |
Reshape | Y | |
Resize | Y | Asymmetric coordinate transformation mode only. Nearest or Linear resizing mode only. "floor" mode only for resize_mode attribute. |
ReverseSequence | Y | |
RNN | Y | |
RoiAlign | N | |
Round | N | |
ScaledTanh | Y | |
Scan | Y | |
Scatter | N | |
ScatterElements | N | |
ScatterND | N | |
Selu | Y | |
SequenceAt | N | |
SequenceConstruct | N | |
SequenceEmpty | N | |
SequenceErase | N | |
SequenceInsert | N | |
SequenceLength | N | |
Shape | Y | |
Shrink | N | |
Sigmoid | Y | |
Sign | N | |
Sin | Y | |
Sinh | Y | |
Size | Y | |
Slice | Y | axes must be an initializer |
Softmax | Y | |
SoftmaxCrossEntropyLoss | Y | |
Softplus | Y | |
Softsign | Y | |
SpaceToDepth | Y | |
Split | Y | split must be an initializer |
SplitToSequence | N | |
Sqrt | Y | |
Squeeze | Y | axes must be an initializer |
StringNormalizer | N | |
Sub | Y | |
Sum | Y | |
Tan | Y | |
Tanh | Y | |
TfIdfVectorizer | N | |
ThresholdedRelu | Y | |
Tile | Y | |
TopK | Y | |
Transpose | Y | |
Unique | N | |
Unsqueeze | Y | axes must be a constant tensor |
Upsample | Y | |
Where | Y | |
Xor | N |