SupportedOperators.md

Supported ONNX operators

Barracuda currently supports the following ONNX operators and parameters. If an operator is not on the list and you need it, please create a ticket on the Unity ML-Agents GitHub.

Operations

• Constant
• Reshape
• Shape
• Unsqueeze
• Squeeze
• Flatten
• Concat
• Slice
• Gather
• OneHot
• LSTM
• Add
• Sum
• Sub
• Mul
• Div
• Pow
• Min
• Max
• Mean
• Greater
• Equal
• Or
• And
• Not
• Xor
• Pad
• AveragePool
• MaxPool
• GlobalAveragePool
• GlobalMaxPool
• Upsample
• Resize
• Transpose
• Gemm
• MatMul
• Conv
• ConvTranspose
• BatchNormalization
• ImageScaler
• InstanceNormalization
• RandomNormal
• RandomNormalLike
• RandomUniform
• RandomUniformLike
• Multinomial
• ReduceMax
• ReduceMean
• ReduceMin
• ReduceProd
• ReduceSum
• Identity
• Cast
• Dropout

Activations

• Relu
• Softmax
• LogSoftmax
• Tanh
• Sqrt
• Sigmoid
• Elu
• LeakyRelu
• Selu
• PRelu
• Exp
• Log
• Reciprocal
• Abs
• Neg
• Ceil
• Floor
• Clip

---

Constant

sparse_value : not supported

Unsqueeze

axis <= 1 : not supported

Squeeze

axis <= 1 : not supported

OneHot

axis : not supported

AveragePool

ceil_mode : not supported

count_include_pad : not supported

MaxPool

ceil_mode : not supported

dilations : not supported

storage_order : not supported

Resize

opset-11 : not supported

=>

coordinate_transformation_mode : not supported

cubic_coeff_a : not supported, default to -0.75

exclude_outside : not supported, default to 0

extrapolation_value : not supported, default to 0

nearest_mode : not supported

Gemm

alpha : not supported, default to 1

beta : not supported, default to 1

transA : not supported, default to 0

ConvTranspose

dilations : not supported, default to {1,1}

group : not supported, default to 1

output_shape : not supported, default to [0]

Softmax

axis : not supported

LogSoftmax

axis : not supported