1. 前言¶
1.1. 版本记录¶
版本 |
修改内容 |
修改时间 |
|---|---|---|
V1.0 |
初版 |
2022-01-12 |
2. TensorFlow 算子支持列表¶
算子 |
支持情况 |
|---|---|
DummyDuplicateOp |
支持 |
DummyDuplicateOp |
支持 |
DummyInfeasibleAttrConstraintOp |
支持 |
DummyReadResource |
支持 |
DummyDuplicateOp |
支持 |
DummyDuplicateOp |
支持 |
DummySideEffectingOp |
支持 |
RandomUniform |
支持 |
RandomShuffle |
支持 |
RandomUniformInt |
支持 |
RandomStandardNormal |
支持 |
TruncatedNormal |
支持 |
ParameterizedTruncatedNormal |
支持 |
RGBToHSV |
支持 |
HSVToRGB |
支持 |
AdjustContrastv2 |
支持 |
AdjustSaturation |
支持 |
AdjustHue |
支持 |
XlaSelfAdjointEig |
支持 |
SelfAdjointEigV2 |
支持 |
ResourceApplyProximalGradientDescent |
支持 |
ResourceApplyMomentum |
支持 |
ResourceApplyAdagrad |
支持 |
ResourceApplyAdagradV2 |
支持 |
ResourceApplyAdagradDA |
支持 |
ResourceApplyAdam |
支持 |
ResourceApplyAdaMax |
支持 |
ResourceApplyRMSProp |
支持 |
ResourceApplyFtrl |
支持 |
ResourceApplyFtrlV2 |
支持 |
ResourceApplyAdadelta |
支持 |
ResourceApplyAddSign |
支持 |
ResourceApplyPowerSign |
支持 |
QuantizeAndDequantizeV2 |
支持 |
QuantizeAndDequantizeV3 |
支持 |
Multinomial |
支持 |
StatelessMultinomial |
支持 |
FusedBatchNorm |
支持 |
FusedBatchNormV2 |
支持 |
FusedBatchNormV3 |
支持 |
_FusedBatchNormEx |
支持 |
FusedBatchNormGrad |
支持 |
FusedBatchNormGradV2 |
支持 |
FusedBatchNormGradV3 |
支持 |
MatrixDiag |
支持 |
MatrixDiagV2 |
支持 |
MatrixDiagPart |
支持 |
MatrixDiagPartV2 |
支持 |
MatrixSetDiag |
支持 |
MatrixSetDiagV2 |
支持 |
Concat |
支持 |
ConcatV2 |
支持 |
ConcatOffset |
支持 |
ListDiff |
支持 |
Reverse |
支持 |
ReverseV2 |
支持 |
MatrixBandPart |
支持 |
Split |
支持 |
SplitV |
支持 |
Elu |
支持 |
EluGrad |
支持 |
Selu |
支持 |
SeluGrad |
支持 |
Range |
支持 |
LinSpace |
支持 |
MaxPool |
支持 |
MaxPoolV2 |
支持 |
MaxPool3D |
支持 |
AvgPool |
支持 |
AvgPool3D |
支持 |
MaxPoolGrad |
支持 |
MaxPoolGradV2 |
支持 |
MaxPool3DGrad |
支持 |
MaxPoolGradGrad |
支持 |
MaxPoolGradGradV2 |
支持 |
MaxPool3DGradGrad |
支持 |
Cross |
支持 |
XlaDynamicUpdateSlice |
支持 |
Slice |
支持 |
XlaConv |
支持 |
ExtractImagePatches |
支持 |
SparseToDense |
支持 |
_UnaryOpsComposition |
支持 |
TensorArrayV3 |
支持 |
TensorArrayWriteV3 |
支持 |
TensorArrayReadV3 |
支持 |
TensorArrayGatherV3 |
支持 |
TensorArrayScatterV3 |
支持 |
TensorArrayConcatV3 |
支持 |
TensorArraySplitV3 |
支持 |
TensorArraySizeV3 |
支持 |
TensorArrayGradV3 |
支持 |
TensorArrayCloseV3 |
支持 |
XlaReplicaId |
支持 |
FakeParam |
支持 |
XlaSelectAndScatter |
支持 |
MirrorPad |
支持 |
FakeQuantWithMinMaxArgs |
支持 |
FakeQuantWithMinMaxArgsGradient |
支持 |
FakeQuantWithMinMaxVars |
支持 |
FakeQuantWithMinMaxVarsGradient |
支持 |
If |
支持 |
StatelessIf |
支持 |
XlaIf |
支持 |
XlaPad |
支持 |
IdentityN |
支持 |
PlaceholderWithDefault |
支持 |
PreventGradient |
支持 |
StopGradient |
支持 |
Snapshot |
支持 |
DynamicStitch |
支持 |
XlaSend |
支持 |
XlaRecv |
支持 |
ArgMax |
支持 |
ArgMin |
支持 |
BroadcastArgs |
支持 |
BroadcastGradientArgs |
支持 |
SpaceToBatchND |
支持 |
SpaceToBatch |
支持 |
Pack |
支持 |
TopKV2 |
支持 |
OneHot |
支持 |
XlaEinsum |
支持 |
StackPushV2 |
支持 |
StackPopV2 |
支持 |
StackCloseV2 |
支持 |
InTopKV2 |
支持 |
While |
支持 |
StatelessWhile |
支持 |
XlaWhile |
支持 |
XlaDot |
支持 |
Sum |
支持 |
Prod |
支持 |
Min |
支持 |
Max |
支持 |
Mean |
支持 |
All |
支持 |
Any |
支持 |
LRN |
支持 |
LRNGrad |
支持 |
SpaceToDepth |
支持 |
Reshape |
支持 |
Cast |
支持 |
Bitcast |
支持 |
Select |
支持 |
SelectV2 |
支持 |
MatrixInverse |
支持 |
Diag |
支持 |
DiagPart |
支持 |
Relu |
支持 |
Relu6 |
支持 |
LeakyRelu |
支持 |
ReluGrad |
支持 |
Relu6Grad |
支持 |
LeakyReluGrad |
支持 |
BroadcastTo |
支持 |
StatelessRandomUniform |
支持 |
StatelessRandomUniformInt |
支持 |
StatelessRandomNormal |
支持 |
StatelessTruncatedNormal |
支持 |
Roll |
支持 |
StatefulUniform |
支持 |
StatefulStandardNormalV2 |
支持 |
StatefulTruncatedNormal |
支持 |
StatefulUniformInt |
支持 |
StatefulUniformFullInt |
支持 |
MatrixTriangularSolve |
支持 |
StridedSlice |
支持 |
StridedSliceGrad |
支持 |
ResourceStridedSliceAssign |
支持 |
FFT |
支持 |
FFT2D |
支持 |
FFT3D |
支持 |
IFFT |
支持 |
IFFT2D |
支持 |
IFFT3D |
支持 |
RFFT |
支持 |
RFFT2D |
支持 |
RFFT3D |
支持 |
IRFFT |
支持 |
IRFFT2D |
支持 |
IRFFT3D |
支持 |
_ListToArray |
支持 |
_ArrayToList |
支持 |
SymbolicGradient |
支持 |
PartitionedCall |
支持 |
StatefulPartitionedCall |
支持 |
Case |
支持 |
VarIsInitializedOp |
支持 |
VariableShape |
支持 |
ReadVariableOp |
支持 |
AssignVariableOp |
支持 |
ResourceGather |
支持 |
ResourceScatterAdd |
支持 |
ResourceScatterSub |
支持 |
ResourceScatterMul |
支持 |
ResourceScatterDiv |
支持 |
ResourceScatterMin |
支持 |
ResourceScatterMax |
支持 |
ResourceScatterUpdate |
支持 |
ResourceScatterNdUpdate |
支持 |
ResourceScatterNdAdd |
支持 |
ResourceScatterNdSub |
支持 |
XlaReduceWindow |
支持 |
Conv2D |
支持 |
Conv3D |
支持 |
DepthwiseConv2dNative |
支持 |
DepthwiseConv2dNativeBackpropInput |
支持 |
DepthwiseConv2dNativeBackpropFilter |
支持 |
Softmax |
支持 |
LogSoftmax |
支持 |
SoftmaxCrossEntropyWithLogits |
支持 |
SparseSoftmaxCrossEntropyWithLogits |
支持 |
XlaReduce |
支持 |
NextAfter |
支持 |
AddN |
支持 |
Qr |
支持 |
Bucketize |
支持 |
Const |
支持 |
ReverseSequence |
支持 |
Assert |
支持 |
XlaDequantize |
支持 |
TensorListLength |
支持 |
TensorListReserve |
支持 |
EmptyTensorList |
支持 |
TensorListElementShape |
支持 |
TensorListGetItem |
支持 |
TensorListGather |
支持 |
TensorListStack |
支持 |
TensorListSetItem |
支持 |
TensorListPushBack |
支持 |
TensorListPopBack |
支持 |
Gather |
支持 |
GatherV2 |
支持 |
GatherNd |
支持 |
ResizeNearestNeighbor |
支持 |
ResizeBilinear |
支持 |
ResizeBilinearGrad |
支持 |
XlaSvd |
支持 |
Svd |
支持 |
Cholesky |
支持 |
Pad |
支持 |
PadV2 |
支持 |
Fill |
支持 |
Resampler |
支持 |
ResamplerGrad |
支持 |
XlaSort |
支持 |
XlaKeyValueSort |
支持 |
CheckNumerics |
支持 |
NoOp |
支持 |
ControlTrigger |
支持 |
ScatterNd |
支持 |
TensorScatterAdd |
支持 |
TensorScatterSub |
支持 |
TensorScatterUpdate |
支持 |
Empty |
支持 |
BiasAdd |
支持 |
BiasAddV1 |
支持 |
BiasAddGrad |
支持 |
Transpose |
支持 |
ConjugateTranspose |
支持 |
InvertPermutation |
支持 |
Cumsum |
支持 |
Cumprod |
支持 |
Unpack |
支持 |
MatMul |
支持 |
SparseMatMul |
支持 |
Shape |
支持 |
ShapeN |
支持 |
Rank |
支持 |
Size |
支持 |
ExpandDims |
支持 |
Squeeze |
支持 |
ZerosLike |
支持 |
OnesLike |
支持 |
Tile |
支持 |
ApproximateEqual |
支持 |
BatchMatMul |
支持 |
BatchMatMulV2 |
支持 |
ClipByValue |
支持 |
BatchToSpaceND |
支持 |
BatchToSpace |
支持 |
DepthToSpace |
支持 |
L2Loss |
支持 |
Add |
支持 |
VarHandleOp |
支持 |
CompilableOp |
支持 |
Add |
支持 |
Sub |
支持 |
Mul |
支持 |
Div |
支持 |
Atan2 |
支持 |
Complex |
支持 |
DivNoNan |
支持 |
MulNoNan |
支持 |
FloorDiv |
支持 |
Xlogy |
支持 |
Xdivy |
支持 |
FloorMod |
支持 |
BitwiseAnd |
支持 |
BitwiseOr |
支持 |
BitwiseXor |
支持 |
LeftShift |
支持 |
RightShift |
支持 |
LogicalAnd |
支持 |
LogicalOr |
支持 |
Mod |
支持 |
Maximum |
支持 |
Minimum |
支持 |
RealDiv |
支持 |
ReciprocalGrad |
支持 |
RsqrtGrad |
支持 |
SqrtGrad |
支持 |
SquaredDifference |
支持 |
TruncateDiv |
支持 |
TruncateMod |
支持 |
Equal |
支持 |
NotEqual |
支持 |
Greater |
支持 |
GreaterEqual |
支持 |
Less |
支持 |
LessEqual |
支持 |
SigmoidGrad |
支持 |
SoftplusGrad |
支持 |
SoftsignGrad |
支持 |
TanhGrad |
支持 |
Pow |
支持 |
NextAfter |
支持 |
ComplexAbs |
支持 |
Angle |
支持 |
Conj |
支持 |
Abs |
支持 |
Acos |
支持 |
Acosh |
支持 |
Asin |
支持 |
Asinh |
支持 |
Atan |
支持 |
Atanh |
支持 |
Ceil |
支持 |
Cos |
支持 |
Cosh |
支持 |
Sin |
支持 |
Exp |
支持 |
Expm1 |
支持 |
Floor |
支持 |
IsFinite |
支持 |
IsInf |
支持 |
IsNan |
支持 |
Inv |
支持 |
Reciprocal |
支持 |
Log |
支持 |
Log1p |
支持 |
Invert |
支持 |
LogicalNot |
支持 |
Neg |
支持 |
Rint |
支持 |
Round |
支持 |
Rsqrt |
支持 |
Sigmoid |
支持 |
Sign |
支持 |
Sinh |
支持 |
Softplus |
支持 |
Softsign |
支持 |
Sqrt |
支持 |
Square |
支持 |
Tan |
支持 |
Tanh |
支持 |
Real |
支持 |
Imag |
支持 |
Erf |
支持 |
Erfc |
支持 |
Lgamma |
支持 |
Digamma |
支持 |