burn/crates/burn-import/SUPPORTED-ONNX-OPS.md

Supported ONNX Operators

Note: some ONNX Ops listed below are pseudo Ops, such as Linear, Conv1d, Conv2d (or other with 1d, 2d suffixes used signify the dimensionality). These are not real ONNX Ops, but are used to represent the corresponding Burn Op.

ONNX OP	Import Support	Burn Support
Abs	✅	✅
Acos	❌	❌
Acosh	❌	❌
Add	✅	✅
And	❌	❌
ArgMax	✅	✅
ArgMin	❌	❌
Asin	❌	❌
Asinh	❌	❌
Atan	❌	❌
Atanh	❌	❌
AveragePool1d	✅	✅
AveragePool2d	✅	✅
BatchNormalization	✅	✅
Bernoulli	❌	❌
BitShift	❌	❌
BitwiseAnd	❌	❌
BitwiseNot	❌	❌
BitwiseOr	❌	❌
BitwiseXor	❌	❌
BlackmanWindow	❌	❌
Cast	✅	✅
CastLike	❌	❌
Ceil	❌	❌
Celu	❌	❌
CenterCropPad	❌	❌
Clip	✅	✅
Col2Im	❌	❌
Compress	❌	❌
Concat	✅	✅
ConcatFromSequence	❌	❌
Constant	✅	✅
ConstantOfShape	✅	✅
Conv1d	✅	✅
Conv2d	✅	✅
Conv3d	✅	✅
ConvInteger	❌	❌
ConvTranspose1d	❌	✅
ConvTranspose2d	✅	✅
ConvTranspose3d	✅	✅
Cos	✅	✅
Cosh	❌	❌
CumSum	❌	❌
DepthToSpace	❌	❌
DequantizeLinear	❌	❌
Det	❌	❌
DFT	❌	❌
Div	✅	✅
Dropout	✅	✅
DynamicQuantizeLinear	❌	❌
Einsum	❌	❌
Elu	❌	❌
Equal	✅	✅
Erf	✅	✅
Exp	✅	✅
Expand	✅	✅
EyeLike	❌	❌
Flatten	✅	✅
Floor	❌	❌
Gather	✅	✅
GatherElements	✅	✅
GatherND	❌	❌
Gelu	✅	✅
Gemm	❌	❌
GlobalAveragePool	✅	✅
GlobalLpPool	❌	❌
GlobalMaxPool	❌	❌
Greater	✅	✅
GreaterOrEqual	✅	✅
GridSample	❌	❌
GroupNormalization	❌	✅
GRU	❌	✅
HammingWindow	❌	❌
HannWindow	❌	❌
Hardmax	❌	❌
HardSigmoid	✅	✅
HardSwish	❌	❌
Identity	✅	✅
If	❌	✅
Im	❌	❌
InstanceNormalization	❌	✅
IsInf	❌	❌
IsNaN	❌	❌
LayerNormalization	✅	✅
LeakyRelu	✅	✅
Less	✅	✅
LessOrEqual	✅	✅
Linear	✅	✅
Log	✅	✅
LogSoftmax	✅	✅
Loop	❌	❌
LpNormalization	❌	❌
LpPool	❌	❌
LRN	❌	❌
LSTM	❌	✅
MatMul	✅	✅
MatMulInteger	❌	✅
Max	✅	✅
MaxPool1d	✅	✅
MaxPool2d	✅	✅
MaxRoiPool	❌	❌
MaxUnpool	❌	❌
Mean	✅	✅
MeanVarianceNormalization	❌	❌
MelWeightMatrix	❌	❌
Min	✅	✅
Mish	❌	❌
Mod	❌	❌
Mul	✅	✅
Multinomial	❌	❌
Neg	✅	✅
NegativeLogLikelihoodLoss	❌	❌
NonMaxSuppression	❌	❌
NonZero	❌	❌
Not	✅	✅
OneHot	❌	✅
Optional	❌	❌
OptionalGetElement	❌	❌
OptionalHasElement	❌	❌
Or	❌	❌
Pad	❌	✅
Pow	✅	✅
PRelu	✅	✅
QLinearConv	❌	❌
QLinearMatMul	❌	❌
QuantizeLinear	❌	❌
RandomNormal	✅	✅
RandomNormalLike	❌	✅
RandomUniform	✅	✅
RandomUniformLike	❌	✅
Range	✅	✅
Reciprocal	✅	✅
ReduceL	❌	❌
ReduceLogSum	❌	❌
ReduceLogSumExp	❌	❌
ReduceMax	✅	✅
ReduceMean	✅	✅
ReduceMin	✅	✅
ReduceProd	✅	✅
ReduceSum	✅	✅
ReduceSumSquare	❌	❌
Relu	✅	✅
Reshape	✅	✅
Resize	✅	✅
ReverseSequence	❌	❌
RNN	❌	✅
RoiAlign	❌	❌
Round	❌	❌
Scan	❌	❌
Scatter	❌	✅
ScatterElements	❌	❌
ScatterND	❌	❌
Selu	❌	❌
SequenceAt	❌	❌
SequenceConstruct	❌	❌
SequenceEmpty	❌	❌
SequenceErase	❌	❌
SequenceInsert	❌	❌
SequenceLength	❌	❌
SequenceMap	❌	❌
Shape	✅	✅
Shrink	❌	❌
Sigmoid	✅	✅
Sign	✅	✅
Sin	✅	✅
Sinh	❌	❌
Size	❌	❌
Slice	✅	✅
Softmax	✅	✅
SoftmaxCrossEntropyLoss	❌	❌
Softplus	❌	❌
Softsign	❌	❌
SpaceToDepth	❌	❌
Split	❌	❌
SplitToSequence	❌	❌
Sqrt	✅	✅
Squeeze	✅	✅
STFT	❌	❌
StringNormalizer	❌	❌
Sub	✅	✅
Sum	✅	✅
Tan	❌	❌
Tanh	✅	✅
TfIdfVectorizer	❌	❌
ThresholdedRelu	❌	❌
Tile	✅	✅
TopK	❌	✅
Transpose	✅	✅
Trilu	❌	✅
Unique	❌	❌
Upsample	❌	❌
Where	✅	✅
Xor	❌	❌
Unsqueeze	✅	✅