https://docs.nvidia.com/deeplearning/sdk/tensorrt-support-matrix/index.html#layers-precision-matrix
3.1. TensorFlow Container 19.07-19.08 (TensorFlow 1.14)
Acos
Acosh
ArgMax
ArgMin
Asin
Asinh
Atan
Ceil
ClipByValue
CombinedNonMaxSuppression
Conv2DBackpropInput
Cos
Cosh
DepthToSpace
Elu
Floor
FusedBatchNormV3
GatherV2
LeakyRelu
Pack
Selu
Sin
Sinh
Softplus
Softsign
SpaceToDepth
Split
SquaredDifference
StopGradient
Tanh
Transpose
Unpack
3.2. TensorFlow Container 19.02-19.06 (TensorFlow 1.13)
ExpandDims
FakeQuantWithMinMaxArgs
FakeQuantWithMinMaxVars
Pow
QuantizeAndDequantizeV2
QuantizeAndDequantizeV3
Reshape
Sigmoid
Slice
Sqrt
Square
Squeeze
StridedSlice
Tanh
3.3. TensorFlow Container 18.11-19.01 (TensorFlow 1.12)
Abs
Add
AvgPool
BatchMatMul
BiasAdd
ConcatV2
Const
Conv2D
DepthwiseConv2dNative
Div
Exp
FusedBatchNorm
FusedBatchNormV2
Identity
Log
MatMul
Max
Maximum
MaxPool
Mean
Min
Minimum
Mul
Neg
Pad
Prod
RealDiv
Reciprocal
Relu
Relu6
Rsqrt
Snapshot
Softmax
Sub
Sum
TopKV2
'AI Development > TensorRT' 카테고리의 다른 글
[TensorRT] Windows + TensorRT 설치하기 (4) | 2019.11.11 |
---|---|
[TensorRT] Yolo v3 to onnx (2) | 2019.10.08 |
[TensorRT] Cuda Error in allocate: 2 (0) | 2019.09.19 |
[TensorRT] keras model 에서 node name 확인하는 방법 (1) | 2019.09.19 |
[TensorRT] 마지막 노드 찾기 (0) | 2019.08.03 |