in python/tvm/relax/frontend/onnx/onnx_frontend.py [0:0]
def _get_convert_map():
return {
# defs/experimental
# "Optional": Optional_,
# "OptionalHasElement": OptionalHasElement,
# "OptionalGetElement": OptionalGetElement,
# Binary operators
"Add": Add,
"Sub": Sub,
"Mul": Mul,
"Div": Div,
"Mod": Mod,
"Less": Less,
"LessOrEqual": LessOrEqual,
"Greater": Greater,
"GreaterOrEqual": GreaterOrEqual,
"Equal": Equal,
"BitwiseAnd": BitwiseAnd,
"BitwiseOr": BitwiseOr,
"BitwiseXor": BitwiseXor,
"BitwiseNot": BitwiseNot,
"BitShift": BitShift,
"And": And,
"Or": Or,
"Xor": Xor,
"Not": Not,
# Unary operators
"Log": Log,
"Exp": Exp,
"Acos": Acos,
"Acosh": Acosh,
"Asin": Asin,
"Asinh": Asinh,
"Atan": Atan,
"Atanh": Atanh,
"Cos": Cos,
"Cosh": Cosh,
"Sin": Sin,
"Sinh": Sinh,
"Tan": Tan,
"Tanh": Tanh,
"Neg": Neg,
"Abs": Abs,
"Reciprocal": Reciprocal,
"Floor": Floor,
"Ceil": Ceil,
"Round": Round,
"IsInf": IsInf,
"IsNaN": IsNaN,
"Sqrt": Sqrt,
"Relu": Relu,
"Selu": Selu,
"Mish": Mish,
"Trilu": Trilu,
"PRelu": PRelu,
"LeakyRelu": LeakyRelu,
"ThresholdedRelu": ThresholdedRelu,
"Elu": Elu,
"Gelu": Gelu,
"FastGelu": FastGelu,
"BiasGelu": BiasGelu,
"HardSigmoid": HardSigmoid,
"HardSwish": HardSwish,
"Sign": Sign,
"Softplus": Softplus,
"Softsign": Softsign,
"Shrink": Shrink,
"Erf": Erf,
"Sum": Sum,
"Min": Min,
"Max": Max,
"Mean": Mean,
"Cast": Cast,
"Gemm": Gemm,
"MatMul": MatMul,
# "MatMulInteger": MatMulInteger,
# "MatMulInteger16": MatMulInteger16,
"Reshape": Reshape,
"Sigmoid": Sigmoid,
"Softmax": Softmax,
"LogSoftmax": LogSoftmax,
"Hardmax": Hardmax,
"Transpose": Transpose,
"Unsqueeze": Unsqueeze,
"Where": Where,
"Concat": Concat,
"Clip": Clip,
"Shape": Shape,
"Pow": Pow,
"CumSum": CumSum,
"Squeeze": Squeeze,
"Constant": Constant,
"Gather": Gather,
"GatherElements": GatherElements,
"GatherND": GatherND,
"Scatter": Scatter,
"ScatterElements": ScatterElements,
"ScatterND": ScatterND,
"Compress": Compress,
"Size": Size,
"EyeLike": EyeLike,
# Normalization
"BatchNormalization": BatchNormalization,
"LayerNormalization": LayerNormalization,
"SkipLayerNormalization": SkipLayerNormalization,
"EmbedLayerNormalization": EmbedLayerNormalization,
"InstanceNormalization": InstanceNormalization,
"MeanVarianceNormalization": MeanVarianceNormalization,
# defs/reduction
"ReduceMax": ReduceMax,
"ReduceMin": ReduceMin,
"ReduceSum": ReduceSum,
"ReduceMean": ReduceMean,
"ReduceProd": ReduceProd,
"ReduceLogSumExp": ReduceLogSumExp,
"ReduceLogSum": ReduceLogSum,
"ReduceSumSquare": ReduceSumSquare,
"ReduceL1": ReduceL1,
"ReduceL2": ReduceL2,
"ArgMax": ArgMax,
"ArgMin": ArgMin,
"TopK": TopK,
"Expand": Expand,
"ConstantOfShape": ConstantOfShape,
"Slice": Slice,
"Attention": Attention,
"Pad": Pad,
"Split": Split,
"Tile": Tile,
"AveragePool": AveragePool,
"MaxPool": MaxPool,
"LpPool": LpPool,
"GlobalAveragePool": GlobalAveragePool,
"GlobalMaxPool": GlobalMaxPool,
"GlobalLpPool": GlobalLpPool,
"MaxUnpool": MaxUnpool,
"Conv": Conv,
"ConvTranspose": ConvTranspose,
"Flatten": Flatten,
"Identity": Identity,
"Resize": Resize,
"Einsum": Einsum,
"Range": Range,
"OneHot": OneHot,
"Unique": Unique,
"NonZero": NonZero,
# "If": If,
# "LRN": LRN,
# "MaxRoiPool": MaxRoiPool,
# "RoiAlign": RoiAlign,
# "NonMaxSuppression": NonMaxSuppression,
# "GridSample": GridSample,
# "Upsample": Upsample,
# others
"DepthToSpace": DepthToSpace,
"SpaceToDepth": SpaceToDepth,
# Sequence operators
"SequenceConstruct": SequenceConstruct,
"SequenceEmpty": SequenceEmpty,
"SequenceErase": SequenceErase,
"SequenceInsert": SequenceInsert,
"SequenceLength": SequenceLength,
"ConcatFromSequence": ConcatFromSequence,
"SplitToSequence": SplitToSequence,
"SequenceAt": SequenceAt,
}