-
Notifications
You must be signed in to change notification settings - Fork 7
KInference TensorFlow.js supported operators
cupertank edited this page Jan 16, 2024
·
35 revisions
Common operators:
- Abs
- Acos
- Acosh
- Add
- And
- ArgMax
- ArgMin
- Asin
- Asinh
- Atan
- Atanh
- AveragePool
- BatchNormalization
- Bernoulli
- BitShift
- BitwiseAnd
- BitwiseNot
- BitwiseOr
- BitwiseXor
- Cast
- CastLike
- Ceil
- Celu
- CenterCropPad
- Clip
- Col2Im
- Compress
- Concat
- ConcatFromSequence
- Constant
- ConstantOfShape
- Conv
- ConvInteger
- ConvTranspose
- Cos
- Cosh
- CumSum
- DFT
- DepthToSpace
- DequantizeLinear
- Det
- Div
- Dropout
- Einsum
- Equal
- Erf
- Exp
- Expand
- EyeLike
- Flatten
- Floor
- GRU
- Gather
- GatherElements
- GatherND
- Gemm
- GlobalAveragePool
- GlobalLpPool
- GlobalMaxPool
- Greater
- GridSample
- GroupNormalization
- Hardmax
- Identity
- If
- InstanceNormalization
- IsInf
- IsNaN
- LRN
- LSTM
- LayerNormalization
- Less
- Log
- LogSoftmax
- Loop
- LpNormalization
- LpPool
- MatMul
- MatMulInteger
- Max
- MaxPool
- MaxRoiPool
- MaxUnpool
- Mean
- MelWeightMatrix
- Min
- Mod
- Mul
- Multinomial
- Neg
- NegativeLogLikelihoodLoss
- NonMaxSuppression
- NonZero
- Not
- OneHot
- Optional
- OptionalGetElement
- OptionalHasElement
- Or
- Pad
- Pow
- QLinearConv
- QLinearMatMul
- QuantizeLinear
- RNN
- RandomNormal
- RandomNormalLike
- RandomUniform
- RandomUniformLike
- Reciprocal
- ReduceMax
- ReduceMean
- ReduceMin
- ReduceProd
- ReduceSum
- Reshape
- Resize
- ReverseSequence
- RoiAlign
- Round
- STFT
- Scan
- Scatter
- ScatterElements
- ScatterND
- SequenceAt
- SequenceConstruct
- SequenceEmpty
- SequenceErase
- SequenceInsert
- SequenceLength
- SequenceMap
- Shape
- Sigmoid
- Sign
- Sin
- Sinh
- Size
- Slice
- Softmax
- SoftmaxCrossEntropyLoss
- SpaceToDepth
- Split
- SplitToSequence
- Sqrt
- Squeeze
- StringNormalizer
- Sub
- Sum
- Tan
- Tanh
- TfIdfVectorizer
- Tile
- TopK
- Transpose
- Trilu
- Unique
- Unsqueeze
- Upsample
- Where
- Xor
Functions:
- BlackmanWindow
- DynamicQuantizeLinear
- Elu
- GreaterOrEqual
- HammingWindow
- HannWindow
- HardSigmoid
- HardSwish
- LeakyRelu
- LessOrEqual
- MeanVarianceNormalization
- Mish
- PRelu
- Range
- ReduceL1
- ReduceL2
- ReduceLogSum
- ReduceLogSumExp
- ReduceSumSquare
- Relu
- Selu
- Shrink
- Softplus
- Softsign
- ThresholdedRelu
- Supported operators: