Skip to content

Commit

Permalink
Bump various ops to opset 22, adding bf16 support
Browse files Browse the repository at this point in the history 
Bumped ops:
Acos
Acosh
Asin
Asinh
Atan
Atanh
AveragePool
Bernoulli
Conv
ConvTranspose
Cos
Cosh
DeformConv
Det
Dropout
Elu
EyeLike
GRU
GlobalAveragePool
GlobalMaxPool
HardSigmoid
HardSwish
InstanceNormalization
LSTM
LpNormalization
LpPool
MaxPool
MaxRoiPool
MaxUnpool
Mish
Multinomial
NegativeLogLikelihoodLoss
RNN
RandomNormal
RandomNormalLike
RandomUniform
RandomUniformLike
RoiAlign
Round
Selu
Sin
Sinh
Softplus
Softsign
Tan
ThresholdedRelu

Signed-off-by: Rickert, Jonas <[email protected]>
  • Loading branch information
@jorickert
jorickert committed Jan 30, 2025
1 parent 6d2b1d4 commit e3fdaad
Show file tree
Hide file tree
Showing 4 changed files with 217 additions and 217 deletions.
4 changes: 2 additions & 2 deletions docs/Dialects/onnx.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1114,13 +1114,13 @@ Effects: `MemoryEffects::Effect{}`

| Operand | Description |
| :-----: | ----------- |
| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 8-bit signless integer values or tensor of 16-bit signless integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 8-bit unsigned integer values or tensor of 16-bit unsigned integer values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 1-bit signless integer values or tensor of string type values or tensor of bfloat16 type values or tensor of f8E4M3FN type values or tensor of f8E4M3FNUZ type values or tensor of f8E5M2 type values or tensor of f8E5M2FNUZ type values
| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 8-bit signless integer values or tensor of 16-bit signless integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 8-bit unsigned integer values or tensor of 16-bit unsigned integer values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 1-bit signless integer values or tensor of string type values or tensor of bfloat16 type values or tensor of f8E4M3FN type values or tensor of f8E4M3FNUZ type values or tensor of f8E5M2 type values or tensor of f8E5M2FNUZ type values or tensor of 4-bit unsigned integer values or tensor of 4-bit signless integer values

#### Results:

| Result | Description |
| :----: | ----------- |
| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 8-bit signless integer values or tensor of 16-bit signless integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 8-bit unsigned integer values or tensor of 16-bit unsigned integer values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 1-bit signless integer values or tensor of string type values or tensor of bfloat16 type values or tensor of f8E4M3FN type values or tensor of f8E4M3FNUZ type values or tensor of f8E5M2 type values or tensor of f8E5M2FNUZ type values
| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 8-bit signless integer values or tensor of 16-bit signless integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 8-bit unsigned integer values or tensor of 16-bit unsigned integer values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 1-bit signless integer values or tensor of string type values or tensor of bfloat16 type values or tensor of f8E4M3FN type values or tensor of f8E4M3FNUZ type values or tensor of f8E5M2 type values or tensor of f8E5M2FNUZ type values or tensor of 4-bit unsigned integer values or tensor of 4-bit signless integer values

### `onnx.CategoryMapper` (ONNXCategoryMapperOp)

Expand Down
92 changes: 46 additions & 46 deletions src/Builder/OpBuildTable.inc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5,22 +5,22 @@
//********************************************************

op_dialect_version_map_["Abs"] = {13};
op_dialect_version_map_["Acos"] = {7};
op_dialect_version_map_["Acosh"] = {9};
op_dialect_version_map_["Acos"] = {22};
op_dialect_version_map_["Acosh"] = {22};
op_dialect_version_map_["Adagrad"] = {1};
op_dialect_version_map_["Adam"] = {1};
op_dialect_version_map_["Add"] = {14};
op_dialect_version_map_["And"] = {7};
op_dialect_version_map_["ArgMax"] = {13};
op_dialect_version_map_["ArgMin"] = {13};
op_dialect_version_map_["ArrayFeatureExtractor"] = {1};
op_dialect_version_map_["Asin"] = {7};
op_dialect_version_map_["Asinh"] = {9};
op_dialect_version_map_["Atan"] = {7};
op_dialect_version_map_["Atanh"] = {9};
op_dialect_version_map_["AveragePool"] = {19};
op_dialect_version_map_["Asin"] = {22};
op_dialect_version_map_["Asinh"] = {22};
op_dialect_version_map_["Atan"] = {22};
op_dialect_version_map_["Atanh"] = {22};
op_dialect_version_map_["AveragePool"] = {22};
op_dialect_version_map_["BatchNormalization"] = {15};
op_dialect_version_map_["Bernoulli"] = {15};
op_dialect_version_map_["Bernoulli"] = {22};
op_dialect_version_map_["Binarizer"] = {1};
op_dialect_version_map_["BitShift"] = {11};
op_dialect_version_map_["BitwiseAnd"] = {18};
Expand All @@ -41,60 +41,60 @@ op_dialect_version_map_["Concat"] = {13};
op_dialect_version_map_["ConcatFromSequence"] = {11};
op_dialect_version_map_["Constant"] = {19};
op_dialect_version_map_["ConstantOfShape"] = {20};
op_dialect_version_map_["Conv"] = {11};
op_dialect_version_map_["Conv"] = {22};
op_dialect_version_map_["ConvInteger"] = {10};
op_dialect_version_map_["ConvTranspose"] = {11};
op_dialect_version_map_["Cos"] = {7};
op_dialect_version_map_["Cosh"] = {9};
op_dialect_version_map_["ConvTranspose"] = {22};
op_dialect_version_map_["Cos"] = {22};
op_dialect_version_map_["Cosh"] = {22};
op_dialect_version_map_["Col2Im"] = {18};
op_dialect_version_map_["CumSum"] = {14};
op_dialect_version_map_["DeformConv"] = {19};
op_dialect_version_map_["DeformConv"] = {22};
op_dialect_version_map_["DepthToSpace"] = {13};
op_dialect_version_map_["DequantizeLinear"] = {19};
op_dialect_version_map_["Det"] = {11};
op_dialect_version_map_["Det"] = {22};
op_dialect_version_map_["DFT"] = {20, 17};
op_dialect_version_map_["DictVectorizer"] = {1};
op_dialect_version_map_["Div"] = {14};
op_dialect_version_map_["Dropout"] = {13};
op_dialect_version_map_["Dropout"] = {22};
op_dialect_version_map_["DynamicQuantizeLinear"] = {11};
op_dialect_version_map_["Einsum"] = {12};
op_dialect_version_map_["Elu"] = {6};
op_dialect_version_map_["Elu"] = {22};
op_dialect_version_map_["Equal"] = {19};
op_dialect_version_map_["Erf"] = {13};
op_dialect_version_map_["Exp"] = {13};
op_dialect_version_map_["Expand"] = {13};
op_dialect_version_map_["EyeLike"] = {9};
op_dialect_version_map_["EyeLike"] = {22};
op_dialect_version_map_["FeatureVectorizer"] = {1};
op_dialect_version_map_["Flatten"] = {13};
op_dialect_version_map_["Floor"] = {13};
op_dialect_version_map_["GRU"] = {14};
op_dialect_version_map_["GRU"] = {22};
op_dialect_version_map_["Gather"] = {13};
op_dialect_version_map_["GatherElements"] = {13};
op_dialect_version_map_["GatherND"] = {13};
op_dialect_version_map_["Gelu"] = {20};
op_dialect_version_map_["Gemm"] = {13};
op_dialect_version_map_["GlobalAveragePool"] = {1};
op_dialect_version_map_["GlobalAveragePool"] = {22};
op_dialect_version_map_["GlobalLpPool"] = {2};
op_dialect_version_map_["GlobalMaxPool"] = {1};
op_dialect_version_map_["GlobalMaxPool"] = {22};
op_dialect_version_map_["Gradient"] = {1};
op_dialect_version_map_["Greater"] = {13};
op_dialect_version_map_["GreaterOrEqual"] = {16};
op_dialect_version_map_["GridSample"] = {16};
op_dialect_version_map_["GroupNormalization"] = {21, 18};
op_dialect_version_map_["HammingWindow"] = {17};
op_dialect_version_map_["HannWindow"] = {17};
op_dialect_version_map_["HardSigmoid"] = {6};
op_dialect_version_map_["HardSigmoid"] = {22};
op_dialect_version_map_["Hardmax"] = {13};
op_dialect_version_map_["HardSwish"] = {14};
op_dialect_version_map_["HardSwish"] = {22};
op_dialect_version_map_["Identity"] = {19};
op_dialect_version_map_["If"] = {19};
op_dialect_version_map_["Imputer"] = {1};
op_dialect_version_map_["InstanceNormalization"] = {6};
op_dialect_version_map_["InstanceNormalization"] = {22};
op_dialect_version_map_["IsInf"] = {20};
op_dialect_version_map_["IsNaN"] = {20};
op_dialect_version_map_["LayerNormalization"] = {17};
op_dialect_version_map_["LRN"] = {13};
op_dialect_version_map_["LSTM"] = {14};
op_dialect_version_map_["LSTM"] = {22};
op_dialect_version_map_["LabelEncoder"] = {2};
op_dialect_version_map_["LeakyRelu"] = {16};
op_dialect_version_map_["Less"] = {13};
Expand All @@ -104,25 +104,25 @@ op_dialect_version_map_["LinearRegressor"] = {1};
op_dialect_version_map_["Log"] = {13};
op_dialect_version_map_["LogSoftmax"] = {13};
op_dialect_version_map_["Loop"] = {19};
op_dialect_version_map_["LpNormalization"] = {1};
op_dialect_version_map_["LpPool"] = {18};
op_dialect_version_map_["LpNormalization"] = {22};
op_dialect_version_map_["LpPool"] = {22};
op_dialect_version_map_["MatMul"] = {13};
op_dialect_version_map_["MatMulInteger"] = {10};
op_dialect_version_map_["Max"] = {13};
op_dialect_version_map_["MaxPool"] = {12};
op_dialect_version_map_["MaxRoiPool"] = {1};
op_dialect_version_map_["MaxUnpool"] = {11};
op_dialect_version_map_["MaxPool"] = {22};
op_dialect_version_map_["MaxRoiPool"] = {22};
op_dialect_version_map_["MaxUnpool"] = {22};
op_dialect_version_map_["Mean"] = {13};
op_dialect_version_map_["MeanVarianceNormalization"] = {13};
op_dialect_version_map_["MelWeightMatrix"] = {17};
op_dialect_version_map_["Min"] = {13};
op_dialect_version_map_["Mish"] = {18};
op_dialect_version_map_["Mish"] = {22};
op_dialect_version_map_["Mod"] = {13};
op_dialect_version_map_["Momentum"] = {1};
op_dialect_version_map_["Mul"] = {14};
op_dialect_version_map_["Multinomial"] = {7};
op_dialect_version_map_["Multinomial"] = {22};
op_dialect_version_map_["Neg"] = {13};
op_dialect_version_map_["NegativeLogLikelihoodLoss"] = {13};
op_dialect_version_map_["NegativeLogLikelihoodLoss"] = {22};
op_dialect_version_map_["NonMaxSuppression"] = {11};
op_dialect_version_map_["NonZero"] = {13};
op_dialect_version_map_["Normalizer"] = {1};
Expand All @@ -139,11 +139,11 @@ op_dialect_version_map_["Pow"] = {15};
op_dialect_version_map_["QLinearConv"] = {10};
op_dialect_version_map_["QLinearMatMul"] = {10};
op_dialect_version_map_["QuantizeLinear"] = {19};
op_dialect_version_map_["RNN"] = {14};
op_dialect_version_map_["RandomNormal"] = {1};
op_dialect_version_map_["RandomNormalLike"] = {1};
op_dialect_version_map_["RandomUniform"] = {1};
op_dialect_version_map_["RandomUniformLike"] = {1};
op_dialect_version_map_["RNN"] = {22};
op_dialect_version_map_["RandomNormal"] = {22};
op_dialect_version_map_["RandomNormalLike"] = {22};
op_dialect_version_map_["RandomUniform"] = {22};
op_dialect_version_map_["RandomUniformLike"] = {22};
op_dialect_version_map_["Range"] = {11};
op_dialect_version_map_["Reciprocal"] = {13};
op_dialect_version_map_["ReduceL1"] = {18, 13};
Expand All @@ -160,16 +160,16 @@ op_dialect_version_map_["Relu"] = {14};
op_dialect_version_map_["Reshape"] = {19};
op_dialect_version_map_["Resize"] = {19, 18, 13, 11, 10};
op_dialect_version_map_["ReverseSequence"] = {10};
op_dialect_version_map_["RoiAlign"] = {16};
op_dialect_version_map_["Round"] = {11};
op_dialect_version_map_["RoiAlign"] = {22};
op_dialect_version_map_["Round"] = {22};
op_dialect_version_map_["SVMClassifier"] = {1};
op_dialect_version_map_["SVMRegressor"] = {1};
op_dialect_version_map_["Scaler"] = {1};
op_dialect_version_map_["Scan"] = {19};
op_dialect_version_map_["Scatter"] = {11};
op_dialect_version_map_["ScatterElements"] = {18};
op_dialect_version_map_["ScatterND"] = {18};
op_dialect_version_map_["Selu"] = {6};
op_dialect_version_map_["Selu"] = {22};
op_dialect_version_map_["SequenceAt"] = {11};
op_dialect_version_map_["SequenceConstruct"] = {11};
op_dialect_version_map_["SequenceEmpty"] = {11};
Expand All @@ -181,14 +181,14 @@ op_dialect_version_map_["Shape"] = {19};
op_dialect_version_map_["Shrink"] = {9};
op_dialect_version_map_["Sigmoid"] = {13};
op_dialect_version_map_["Sign"] = {13};
op_dialect_version_map_["Sin"] = {7};
op_dialect_version_map_["Sinh"] = {9};
op_dialect_version_map_["Sin"] = {22};
op_dialect_version_map_["Sinh"] = {22};
op_dialect_version_map_["Size"] = {19};
op_dialect_version_map_["Slice"] = {13};
op_dialect_version_map_["Softmax"] = {13, 11};
op_dialect_version_map_["SoftmaxCrossEntropyLoss"] = {13};
op_dialect_version_map_["Softplus"] = {1};
op_dialect_version_map_["Softsign"] = {1};
op_dialect_version_map_["Softplus"] = {22};
op_dialect_version_map_["Softsign"] = {22};
op_dialect_version_map_["SpaceToDepth"] = {13};
op_dialect_version_map_["Split"] = {18, 13, 11};
op_dialect_version_map_["SplitToSequence"] = {11};
Expand All @@ -198,10 +198,10 @@ op_dialect_version_map_["StringNormalizer"] = {10};
op_dialect_version_map_["STFT"] = {17};
op_dialect_version_map_["Sub"] = {14};
op_dialect_version_map_["Sum"] = {13};
op_dialect_version_map_["Tan"] = {7};
op_dialect_version_map_["Tan"] = {22};
op_dialect_version_map_["Tanh"] = {13};
op_dialect_version_map_["TfIdfVectorizer"] = {9};
op_dialect_version_map_["ThresholdedRelu"] = {10};
op_dialect_version_map_["ThresholdedRelu"] = {22};
op_dialect_version_map_["Tile"] = {13};
op_dialect_version_map_["TopK"] = {11};
op_dialect_version_map_["Transpose"] = {13};
Expand Down
Loading

0 comments on commit e3fdaad

Please sign in to comment.