Supported ONNX operators | Barracuda | 1.0.4
docs.unity.cn
    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 © 2020 Unity Technologies
    Generated by DocFX
    on 20 October 2020