Supported ONNX operators | Barracuda | 0.8.0-preview
docs.unity3d.com
    Show / Hide Table of Contents

    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

    Back to top
    Copyright © 2023 Unity Technologies — Terms of use
    • Legal
    • Privacy Policy
    • Cookies
    • Do Not Sell or Share My Personal Information
    • Your Privacy Choices (Cookie Settings)
    "Unity", Unity logos, and other Unity trademarks are trademarks or registered trademarks of Unity Technologies or its affiliates in the U.S. and elsewhere (more info here). Other names or brands are trademarks of their respective owners.
    Generated by DocFX on 18 October 2023