From c3ad4f3af578c035f7bc57cca4048408ff32303c Mon Sep 17 00:00:00 2001 From: Alexandre Eichenberger Date: Wed, 3 Aug 2022 14:17:01 -0400 Subject: [PATCH 1/5] Print type of ONNX operations's input operand at runtime (#1575) * proper way to do it, with onnx transfo adding an onnx print signature that gets lowered Signed-off-by: Alexandre Eichenberger --- docs/Dialects/krnl.md | 6 + docs/Dialects/onnx.md | 1248 +++++++++-------- docs/Dialects/zhigh.md | 31 + docs/Dialects/zlow.md | 37 + .../NNPA/Compiler/NNPACompilerUtils.cpp | 3 +- src/Compiler/CompilerOptions.cpp | 4 + src/Compiler/CompilerOptions.hpp | 1 + src/Compiler/CompilerPasses.cpp | 9 +- src/Compiler/CompilerPasses.hpp | 4 +- src/Conversion/ONNXToKrnl/CMakeLists.txt | 17 +- .../ONNXToKrnl/ConvertONNXToKrnl.cpp | 1 + .../ONNXToKrnl/ONNXToKrnlCommon.hpp | 2 + .../ONNXToKrnl/Tensor/PrintSignature.cpp | 55 + src/Dialect/Krnl/Krnl.td | 6 + src/Dialect/Krnl/KrnlHelper.cpp | 10 +- src/Dialect/ONNX/AdditionalONNXOps.td | 9 + src/InitOMPasses.hpp | 4 + src/Pass/Passes.hpp | 4 + src/Runtime/OMTensor.inc | 100 +- src/Transform/ONNX/CMakeLists.txt | 1 + .../ONNX/InstrumentONNXSignaturePass.cpp | 99 ++ 21 files changed, 1040 insertions(+), 611 deletions(-) create mode 100644 src/Conversion/ONNXToKrnl/Tensor/PrintSignature.cpp create mode 100644 src/Transform/ONNX/InstrumentONNXSignaturePass.cpp diff --git a/docs/Dialects/krnl.md b/docs/Dialects/krnl.md index 3c64bb2953f..fdaa78fcdbc 100644 --- a/docs/Dialects/krnl.md +++ b/docs/Dialects/krnl.md @@ -939,6 +939,12 @@ Traits: MemRefsNormalizable Print a tensor. This operation can be used to generate a call to a runtime function which prints a tensor. +At the begining of the msg string, user can add formatting instructions. The flags are: + %s: detailed signature (including shape, type, offsets), + %t: compact type (ala MLIR: 32x16xfloat), + %d: data values. +When no formatting is provided, "%s%d" is used (detailed signature and data) by default. +Print operation ends with a newline, except when only requesting a compact types (%t). Traits: MemRefsNormalizable diff --git a/docs/Dialects/onnx.md b/docs/Dialects/onnx.md index 66657bb6194..338d671448a 100644 --- a/docs/Dialects/onnx.md +++ b/docs/Dialects/onnx.md @@ -15,13 +15,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `X` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `Y` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.Acos` (::mlir::ONNXAcosOp) @@ -37,13 +37,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 memref of any type values +| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Acosh` (::mlir::ONNXAcoshOp) @@ -59,13 +59,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 memref of any type values +| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Adagrad` (::mlir::ONNXAdagradOp) @@ -138,15 +138,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `R` | tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `T` | tensor of 64-bit signless integer values or memref of any type values -| `inputs` | tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `R` | tensor of 32-bit float values or tensor of 64-bit float values +| `T` | tensor of 64-bit signless integer values +| `inputs` | tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `outputs` | tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `outputs` | tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Adam` (::mlir::ONNXAdamOp) @@ -232,15 +232,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `R` | tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `T` | tensor of 64-bit signless integer values or memref of any type values -| `inputs` | tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `R` | tensor of 32-bit float values or tensor of 64-bit float values +| `T` | tensor of 64-bit signless integer values +| `inputs` | tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `outputs` | tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `outputs` | tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Add` (::mlir::ONNXAddOp) @@ -260,14 +260,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `A` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values -| `B` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `A` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `B` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `C` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `C` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.And` (::mlir::ONNXAndOp) @@ -286,14 +286,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `A` | tensor of 1-bit signless integer values or memref of any type values -| `B` | tensor of 1-bit signless integer values or memref of any type values +| `A` | tensor of 1-bit signless integer values +| `B` | tensor of 1-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `C` | tensor of 1-bit signless integer values or memref of any type values +| `C` | tensor of 1-bit signless integer values ### `onnx.ArgMax` (::mlir::ONNXArgMaxOp) @@ -323,13 +323,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `data` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `reduced` | tensor of 64-bit signless integer values or memref of any type values +| `reduced` | tensor of 64-bit signless integer values ### `onnx.ArgMin` (::mlir::ONNXArgMinOp) @@ -359,13 +359,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `data` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `reduced` | tensor of 64-bit signless integer values or memref of any type values +| `reduced` | tensor of 64-bit signless integer values ### `onnx.ArrayFeatureExtractor` (::mlir::ONNXArrayFeatureExtractorOp) @@ -382,14 +382,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values or tensor of string type values or memref of any type values -| `Y` | tensor of 64-bit signless integer values or memref of any type values +| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values or tensor of string type values +| `Y` | tensor of 64-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `Z` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values or tensor of string type values or memref of any type values +| `Z` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values or tensor of string type values ### `onnx.Asin` (::mlir::ONNXAsinOp) @@ -405,13 +405,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 memref of any type values +| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Asinh` (::mlir::ONNXAsinhOp) @@ -427,13 +427,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 memref of any type values +| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Atan` (::mlir::ONNXAtanOp) @@ -449,13 +449,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 memref of any type values +| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Atanh` (::mlir::ONNXAtanhOp) @@ -471,13 +471,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 memref of any type values +| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.AveragePool` (::mlir::ONNXAveragePoolOp) @@ -532,13 +532,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.BatchNormalizationInferenceMode` (::mlir::ONNXBatchNormalizationInferenceModeOp) @@ -588,13 +588,42 @@ ONNX BatchNormalization operation "Carries out batch normalization as described in the paper" "https://arxiv.org/abs/1502.03167. Depending on the mode it is being run," -"there are multiple cases for the number of outputs, which we list below:" +"There are five required inputs 'X', 'scale', 'B', 'input_mean' and" +"'input_var'." +"Note that 'input_mean' and 'input_var' are expected to be the estimated" +"statistics in inference mode (training_mode=False, default)," +"and the running statistics in training mode (training_mode=True)." +"There are multiple cases for the number of outputs, which we list below:" +"" +"Output case #1: Y, running_mean, running_var (training_mode=True)" +"Output case #2: Y (training_mode=False)" +"" +"When training_mode=False, extra outputs are invalid." +"The outputs are updated as follows when training_mode=True:" +"```" +"running_mean = input_mean * momentum + current_mean * (1 - momentum)" +"running_var = input_var * momentum + current_var * (1 - momentum)" "" -"Output case #1: Y, mean, var, saved_mean, saved_var (training mode)" -"Output case #2: Y (test mode)" +"Y = (X - current_mean) / sqrt(current_var + epsilon) * scale + B" +"" +"where:" +"" +"current_mean = ReduceMean(X, axis=all_except_channel_index)" +"current_var = ReduceVar(X, axis=all_except_channel_index)" +"" +"Notice that ReduceVar refers to the population variance, and it equals to" +"sum(sqrd(x_i - x_avg)) / N" +"where N is the population size (this formula does not use sample size N - 1)." +"" +"```" +"" +"When training_mode=False:" +"```" +"Y = (X - input_mean) / sqrt(input_var + epsilon) * scale + B" +"```" "" "For previous (depreciated) non-spatial cases, implementors are suggested" -"to flatten the input shape to (N x C*D1*D2 ..*Dn) before a BatchNormalization Op." +"to flatten the input shape to (N x C * D1 * D2 * ... * Dn) before a BatchNormalization Op." "This operator has **optional** inputs/outputs. See [the doc](IR.md) for more details about the representation of optional arguments. An empty string may be used in the place of an actual argument's name to indicate a missing argument. Trailing optional arguments (those not followed by an argument that is present) may also be simply omitted." Interfaces: NoSideEffect (MemoryEffectOpInterface), ShapeInference @@ -607,26 +636,25 @@ Effects: MemoryEffects::Effect{} | :-------: | :-------: | ----------- | | `epsilon` | ::mlir::FloatAttr | 32-bit float attribute | `momentum` | ::mlir::FloatAttr | 32-bit float attribute +| `training_mode` | ::mlir::IntegerAttr | 64-bit signed integer attribute #### Operands: | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `scale` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `B` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `mean` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `var` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `scale` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `B` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `input_mean` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `input_var` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `out_mean` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type -| `out_var` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type -| `saved_mean` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type -| `saved_var` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `running_mean` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or none type +| `running_var` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or none type ### `onnx.Binarizer` (::mlir::ONNXBinarizerOp) @@ -648,13 +676,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values or memref of any type values +| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values or memref of any type values +| `Y` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values ### `onnx.BitShift` (::mlir::ONNXBitShiftOp) @@ -687,14 +715,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | 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 memref of any type values -| `Y` | 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 memref of any type values +| `X` | 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 +| `Y` | 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 #### Results: | Result | Description | | :----: | ----------- | -| `Z` | 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 memref of any type values +| `Z` | 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 ### `onnx.ONNX_Call` (::mlir::ONNXCallOp) @@ -764,13 +792,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tuple with any combination of 64-bit signless integer or string type values or tuple with any combination of 64-bit signless integer or 32-bit float values or memref of any type values +| `X` | tuple with any combination of 64-bit signless integer or string type values or tuple with any combination of 64-bit signless integer or 32-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of string type values or tensor of 32-bit float values or tensor of 64-bit signless integer values or memref of any type values +| `Y` | tensor of string type values or tensor of 32-bit float values or tensor of 64-bit signless integer values ### `onnx.Cast` (::mlir::ONNXCastOp) @@ -810,13 +838,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 memref of any 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 #### 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 memref of any 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 ### `onnx.CategoryMapper` (::mlir::ONNXCategoryMapperOp) @@ -848,13 +876,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of string type values or tensor of 64-bit signless integer values or memref of any type values +| `X` | tensor of string type values or tensor of 64-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of string type values or tensor of 64-bit signless integer values or memref of any type values +| `Y` | tensor of string type values or tensor of 64-bit signless integer values ### `onnx.Ceil` (::mlir::ONNXCeilOp) @@ -872,13 +900,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.Celu` (::mlir::ONNXCeluOp) @@ -906,13 +934,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 32-bit float values or memref of any type values +| `X` | tensor of 32-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 32-bit float values or memref of any type values +| `Y` | tensor of 32-bit float values ### `onnx.Clip` (::mlir::ONNXClipOp) @@ -930,15 +958,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values -| `min` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values or none type -| `max` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values or none type +| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `min` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or none type +| `max` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or none type #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.ClipV11` (::mlir::ONNXClipV11Op) @@ -956,15 +984,15 @@ 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 memref of any type values -| `min` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type -| `max` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type +| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `min` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type +| `max` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type #### 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.ClipV12` (::mlir::ONNXClipV12Op) @@ -982,15 +1010,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `min` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type -| `max` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type +| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `min` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type +| `max` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.ClipV6` (::mlir::ONNXClipV6Op) @@ -1015,13 +1043,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 memref of any type values +| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Compress` (::mlir::ONNXCompressOp) @@ -1046,14 +1074,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `condition` | tensor of 1-bit signless integer values or memref of any type values +| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `condition` | tensor of 1-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.ConcatFromSequence` (::mlir::ONNXConcatFromSequenceOp) @@ -1079,13 +1107,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input_sequence` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or memref of any type values +| `input_sequence` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values #### Results: | Result | Description | | :----: | ----------- | -| `concat_result` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `concat_result` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.Concat` (::mlir::ONNXConcatOp) @@ -1107,13 +1135,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `inputs` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `inputs` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `concat_result` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `concat_result` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.ConstantOfShape` (::mlir::ONNXConstantOfShapeOp) @@ -1135,13 +1163,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | tensor of 64-bit signless integer values or memref of any type values +| `input` | tensor of 64-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 memref of any 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 ### `onnx.Constant` (::mlir::ONNXConstantOp) @@ -1171,7 +1199,7 @@ Effects: MemoryEffects::Effect{} | Result | Description | | :----: | ----------- | -| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.ConvInteger` (::mlir::ONNXConvIntegerOp) @@ -1199,16 +1227,16 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `x` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values -| `w` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values -| `x_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values or none type -| `w_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values or none type +| `x` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values +| `w` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values +| `x_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or none type +| `w_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or none type #### Results: | Result | Description | | :----: | ----------- | -| `y` | tensor of 32-bit signless integer values or memref of any type values +| `y` | tensor of 32-bit signless integer values ### `onnx.Conv` (::mlir::ONNXConvOp) @@ -1236,15 +1264,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `W` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `B` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `W` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `B` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.ConvTranspose` (::mlir::ONNXConvTransposeOp) @@ -1286,15 +1314,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `W` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `B` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `W` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `B` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Cos` (::mlir::ONNXCosOp) @@ -1310,13 +1338,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 memref of any type values +| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Cosh` (::mlir::ONNXCoshOp) @@ -1332,13 +1360,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 memref of any type values +| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.CumSum` (::mlir::ONNXCumSumOp) @@ -1380,14 +1408,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `x` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values -| `axis` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values +| `x` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `axis` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `y` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `y` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.Custom` (::mlir::ONNXCustomOp) @@ -1467,13 +1495,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `input` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.DequantizeLinear` (::mlir::ONNXDequantizeLinearOp) @@ -1499,15 +1527,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `x` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or tensor of 32-bit signless integer values or memref of any type values -| `x_scale` | tensor of 32-bit float values or memref of any type values -| `x_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or tensor of 32-bit signless integer values or memref of any type values or none type +| `x` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or tensor of 32-bit signless integer values +| `x_scale` | tensor of 32-bit float values +| `x_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or tensor of 32-bit signless integer values or none type #### Results: | Result | Description | | :----: | ----------- | -| `y` | tensor of 32-bit float values or memref of any type values +| `y` | tensor of 32-bit float values ### `onnx.Det` (::mlir::ONNXDetOp) @@ -1527,13 +1555,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.DictVectorizer` (::mlir::ONNXDictVectorizerOp) @@ -1567,13 +1595,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tuple with any combination of string type or 64-bit signless integer values or tuple with any combination of 64-bit signless integer or string type values or tuple with any combination of 64-bit signless integer or 32-bit float values or tuple with any combination of 64-bit signless integer or 64-bit float values or tuple with any combination of string type or 32-bit float values or tuple with any combination of string type or 64-bit float values or memref of any type values +| `X` | tuple with any combination of string type or 64-bit signless integer values or tuple with any combination of 64-bit signless integer or string type values or tuple with any combination of 64-bit signless integer or 32-bit float values or tuple with any combination of 64-bit signless integer or 64-bit float values or tuple with any combination of string type or 32-bit float values or tuple with any combination of string type or 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 64-bit signless integer values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or memref of any type values +| `Y` | tensor of 64-bit signless integer values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values ### `onnx.Div` (::mlir::ONNXDivOp) @@ -1593,14 +1621,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `A` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values -| `B` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `A` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `B` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `C` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `C` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.Dropout` (::mlir::ONNXDropoutOp) @@ -1633,16 +1661,16 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values -| `ratio` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type -| `training_mode` | tensor of 1-bit signless integer values or memref of any type values or none type +| `data` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `ratio` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type +| `training_mode` | tensor of 1-bit signless integer values or none type #### 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 bfloat16 type values or memref of any type values -| `mask` | tensor of 1-bit signless integer values or memref of any type values or none type +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `mask` | tensor of 1-bit signless integer values or none type ### `onnx.DynamicQuantizeLinear` (::mlir::ONNXDynamicQuantizeLinearOp) @@ -1679,15 +1707,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `x` | tensor of 32-bit float values or memref of any type values +| `x` | tensor of 32-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `y` | tensor of 8-bit unsigned integer values or memref of any type values -| `y_scale` | tensor of 32-bit float values or memref of any type values -| `y_zero_point` | tensor of 8-bit unsigned integer values or memref of any type values +| `y` | tensor of 8-bit unsigned integer values +| `y_scale` | tensor of 32-bit float values +| `y_zero_point` | tensor of 8-bit unsigned integer values ### `onnx.Einsum` (::mlir::ONNXEinsumOp) @@ -1731,13 +1759,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `Inputs` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Inputs` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Elu` (::mlir::ONNXEluOp) @@ -1762,13 +1790,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.EntryPoint` (::mlir::ONNXEntryPointOp) @@ -1799,14 +1827,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `A` | tensor of 1-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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values -| `B` | tensor of 1-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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `A` | tensor of 1-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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `B` | tensor of 1-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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `C` | tensor of 1-bit signless integer values or memref of any type values +| `C` | tensor of 1-bit signless integer values ### `onnx.Erf` (::mlir::ONNXErfOp) @@ -1822,13 +1850,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.Exp` (::mlir::ONNXExpOp) @@ -1844,13 +1872,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 bfloat16 type values or memref of any 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 bfloat16 type 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 bfloat16 type values or memref of any 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 bfloat16 type values ### `onnx.Expand` (::mlir::ONNXExpandOp) @@ -1873,14 +1901,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `shape` | tensor of 64-bit signless integer values or memref of any type values +| `input` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `shape` | tensor of 64-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.EyeLike` (::mlir::ONNXEyeLikeOp) @@ -1909,13 +1937,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 memref of any 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 #### 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 memref of any 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 ### `onnx.FeatureVectorizer` (::mlir::ONNXFeatureVectorizerOp) @@ -1940,13 +1968,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `X` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 32-bit float values or memref of any type values +| `Y` | tensor of 32-bit float values ### `onnx.Flatten` (::mlir::ONNXFlattenOp) @@ -1970,13 +1998,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `input` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.Floor` (::mlir::ONNXFloorOp) @@ -1994,13 +2022,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.GRU` (::mlir::ONNXGRUOp) @@ -2094,25 +2122,26 @@ Effects: MemoryEffects::Effect{} | `clip` | ::mlir::FloatAttr | 32-bit float attribute | `direction` | ::mlir::StringAttr | string attribute | `hidden_size` | ::mlir::IntegerAttr | 64-bit signed integer attribute +| `layout` | ::mlir::IntegerAttr | 64-bit signed integer attribute | `linear_before_reset` | ::mlir::IntegerAttr | 64-bit signed integer attribute #### Operands: | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `W` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `R` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `B` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type -| `sequence_lens` | tensor of 32-bit signless integer values or memref of any type values or none type -| `initial_h` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `W` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `R` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `B` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type +| `sequence_lens` | tensor of 32-bit signless integer values or none type +| `initial_h` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type -| `Y_h` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type +| `Y_h` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type ### `onnx.GatherElements` (::mlir::ONNXGatherElementsOp) @@ -2184,14 +2213,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `indices` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values +| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `indices` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.GatherND` (::mlir::ONNXGatherNDOp) @@ -2303,14 +2332,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `indices` | tensor of 64-bit signless integer values or memref of any type values +| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `indices` | tensor of 64-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.Gather` (::mlir::ONNXGatherOp) @@ -2386,14 +2415,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `indices` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values +| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `indices` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.Gemm` (::mlir::ONNXGemmOp) @@ -2430,15 +2459,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `A` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of bfloat16 type values or memref of any type values -| `B` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of bfloat16 type values or memref of any type values -| `C` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of bfloat16 type values or memref of any type values or none type +| `A` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of bfloat16 type values +| `B` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of bfloat16 type values +| `C` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of bfloat16 type values or none type #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of bfloat16 type values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of bfloat16 type values ### `onnx.GlobalAveragePool` (::mlir::ONNXGlobalAveragePoolOp) @@ -2456,13 +2485,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.GlobalLpPool` (::mlir::ONNXGlobalLpPoolOp) @@ -2486,13 +2515,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.GlobalMaxPool` (::mlir::ONNXGlobalMaxPoolOp) @@ -2510,13 +2539,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Gradient` (::mlir::ONNXGradientOp) @@ -2662,13 +2691,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `Inputs` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `Inputs` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `Outputs` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Outputs` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Greater` (::mlir::ONNXGreaterOp) @@ -2687,14 +2716,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `A` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values -| `B` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `A` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `B` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `C` | tensor of 1-bit signless integer values or memref of any type values +| `C` | tensor of 1-bit signless integer values ### `onnx.GreaterOrEqual` (::mlir::ONNXGreaterOrEqualOp) @@ -2713,14 +2742,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `A` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values -| `B` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `A` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `B` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `C` | tensor of 1-bit signless integer values or memref of any type values +| `C` | tensor of 1-bit signless integer values ### `onnx.HardSigmoid` (::mlir::ONNXHardSigmoidOp) @@ -2745,13 +2774,37 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values + +### `onnx.HardSwish` (::mlir::ONNXHardSwishOp) + +ONNX HardSwish operation + +"HardSwish takes one input data (Tensor) and produces one output data (Tensor) where" +"the HardSwish function, y = x * max(0, min(1, alpha * x + beta)) = x * HardSigmoid(x)," +"where alpha = 1/6 and beta = 0.5, is applied to the tensor elementwise." + +Interfaces: NoSideEffect (MemoryEffectOpInterface), ShapeInference + +Effects: MemoryEffects::Effect{} + +#### Operands: + +| Operand | Description | +| :-----: | ----------- | +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values + +#### Results: + +| Result | Description | +| :----: | ----------- | +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Hardmax` (::mlir::ONNXHardmaxOp) @@ -2779,13 +2832,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 bfloat16 type values or memref of any 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 bfloat16 type 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 bfloat16 type values or memref of any 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 bfloat16 type values ### `onnx.Identity` (::mlir::ONNXIdentityOp) @@ -2801,13 +2854,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `input` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.If` (::mlir::ONNXIfOp) @@ -2823,13 +2876,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `cond` | tensor of 1-bit signless integer values or memref of any type values +| `cond` | tensor of 1-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `outputs` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of bfloat16 type values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of bfloat16 type values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `outputs` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of bfloat16 type values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of bfloat16 type values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.Imputer` (::mlir::ONNXImputerOp) @@ -2861,13 +2914,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values or memref of any type values +| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values or memref of any type values +| `Y` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values ### `onnx.InstanceNormalization` (::mlir::ONNXInstanceNormalizationOp) @@ -2894,15 +2947,15 @@ 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 memref of any type values -| `scale` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `B` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `scale` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `B` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.IsInf` (::mlir::ONNXIsInfOp) @@ -2925,13 +2978,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `X` | tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 1-bit signless integer values or memref of any type values +| `Y` | tensor of 1-bit signless integer values ### `onnx.IsNaN` (::mlir::ONNXIsNaNOp) @@ -2947,13 +3000,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 1-bit signless integer values or memref of any type values +| `Y` | tensor of 1-bit signless integer values ### `onnx.LRN` (::mlir::ONNXLRNOp) @@ -2987,13 +3040,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.LSTM` (::mlir::ONNXLSTMOp) @@ -3096,27 +3149,28 @@ Effects: MemoryEffects::Effect{} | `direction` | ::mlir::StringAttr | string attribute | `hidden_size` | ::mlir::IntegerAttr | 64-bit signed integer attribute | `input_forget` | ::mlir::IntegerAttr | 64-bit signed integer attribute +| `layout` | ::mlir::IntegerAttr | 64-bit signed integer attribute #### Operands: | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `W` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `R` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `B` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type -| `sequence_lens` | tensor of 32-bit signless integer values or memref of any type values or none type -| `initial_h` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type -| `initial_c` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type -| `P` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `W` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `R` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `B` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type +| `sequence_lens` | tensor of 32-bit signless integer values or none type +| `initial_h` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type +| `initial_c` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type +| `P` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type -| `Y_h` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type -| `Y_c` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type +| `Y_h` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type +| `Y_c` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type ### `onnx.LabelEncoder` (::mlir::ONNXLabelEncoderOp) @@ -3162,13 +3216,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of string type values or tensor of 64-bit signless integer values or tensor of 32-bit float values or memref of any type values +| `X` | tensor of string type values or tensor of 64-bit signless integer values or tensor of 32-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of string type values or tensor of 64-bit signless integer values or tensor of 32-bit float values or memref of any type values +| `Y` | tensor of string type values or tensor of 64-bit signless integer values or tensor of 32-bit float values ### `onnx.LeakyRelu` (::mlir::ONNXLeakyReluOp) @@ -3195,13 +3249,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `X` | tensor of bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Y` | tensor of bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Less` (::mlir::ONNXLessOp) @@ -3220,14 +3274,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `A` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values -| `B` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `A` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `B` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `C` | tensor of 1-bit signless integer values or memref of any type values +| `C` | tensor of 1-bit signless integer values ### `onnx.LessOrEqual` (::mlir::ONNXLessOrEqualOp) @@ -3246,14 +3300,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `A` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values -| `B` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `A` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `B` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `C` | tensor of 1-bit signless integer values or memref of any type values +| `C` | tensor of 1-bit signless integer values ### `onnx.LinearClassifier` (::mlir::ONNXLinearClassifierOp) @@ -3280,14 +3334,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values or memref of any type values +| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of string type values or tensor of 64-bit signless integer values or memref of any type values -| `Z` | tensor of 32-bit float values or memref of any type values +| `Y` | tensor of string type values or tensor of 64-bit signless integer values +| `Z` | tensor of 32-bit float values ### `onnx.LinearRegressor` (::mlir::ONNXLinearRegressorOp) @@ -3317,13 +3371,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values or memref of any type values +| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 32-bit float values or memref of any type values +| `Y` | tensor of 32-bit float values ### `onnx.Log` (::mlir::ONNXLogOp) @@ -3339,13 +3393,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 bfloat16 type values or memref of any 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 bfloat16 type 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 bfloat16 type values or memref of any 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 bfloat16 type values ### `onnx.LogSoftmax` (::mlir::ONNXLogSoftmaxOp) @@ -3373,13 +3427,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 bfloat16 type values or memref of any 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 bfloat16 type 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 bfloat16 type values or memref of any 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 bfloat16 type values ### `onnx.Loop` (::mlir::ONNXLoopOp) @@ -3529,15 +3583,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `M` | tensor of 64-bit signless integer values or memref of any type values or none type -| `cond` | tensor of 1-bit signless integer values or memref of any type values or none type -| `v_initial` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of bfloat16 type values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of bfloat16 type values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `M` | tensor of 64-bit signless integer values or none type +| `cond` | tensor of 1-bit signless integer values or none type +| `v_initial` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of bfloat16 type values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of bfloat16 type values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `v_final_and_scan_outputs` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of bfloat16 type values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of bfloat16 type values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `v_final_and_scan_outputs` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of bfloat16 type values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of bfloat16 type values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.LpNormalization` (::mlir::ONNXLpNormalizationOp) @@ -3560,13 +3614,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 memref of any type values +| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.LpPool` (::mlir::ONNXLpPoolOp) @@ -3596,13 +3650,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.MatMulInteger` (::mlir::ONNXMatMulIntegerOp) @@ -3619,16 +3673,16 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `A` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values -| `B` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values -| `a_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values or none type -| `b_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values or none type +| `A` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values +| `B` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values +| `a_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or none type +| `b_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or none type #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 32-bit signless integer values or memref of any type values +| `Y` | tensor of 32-bit signless integer values ### `onnx.MatMul` (::mlir::ONNXMatMulOp) @@ -3644,14 +3698,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `A` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of bfloat16 type values or memref of any type values -| `B` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of bfloat16 type values or memref of any type values +| `A` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of bfloat16 type values +| `B` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of bfloat16 type values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of bfloat16 type values ### `onnx.Max` (::mlir::ONNXMaxOp) @@ -3669,13 +3723,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data_0` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `data_0` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `max` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `max` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.MaxPool` (::mlir::ONNXMaxPoolOp) @@ -3731,14 +3785,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | 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 8-bit unsigned integer values or memref of any type values +| `X` | 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 8-bit unsigned integer values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | 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 8-bit unsigned integer values or memref of any type values -| `Indices` | tensor of 64-bit signless integer values or memref of any type values or none type +| `Y` | 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 8-bit unsigned integer values +| `Indices` | tensor of 64-bit signless integer values or none type ### `onnx.MaxPoolSingleOut` (::mlir::ONNXMaxPoolSingleOutOp) @@ -3798,14 +3852,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `rois` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `rois` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.MaxUnpool` (::mlir::ONNXMaxUnpoolOp) @@ -3846,15 +3900,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `I` | tensor of 64-bit signless integer values or memref of any type values -| `output_shape` | tensor of 64-bit signless integer values or memref of any type values or none type +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `I` | tensor of 64-bit signless integer values +| `output_shape` | tensor of 64-bit signless integer values or none type #### 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Mean` (::mlir::ONNXMeanOp) @@ -3872,13 +3926,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data_0` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `data_0` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `mean` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `mean` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.MeanVarianceNormalization` (::mlir::ONNXMeanVarianceNormalizationOp) @@ -3901,13 +3955,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.Min` (::mlir::ONNXMinOp) @@ -3925,13 +3979,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data_0` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `data_0` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `min` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `min` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.Mod` (::mlir::ONNXModOp) @@ -3965,14 +4019,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `A` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values -| `B` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `A` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `B` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `C` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `C` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.Momentum` (::mlir::ONNXMomentumOp) @@ -4056,15 +4110,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `R` | tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `T` | tensor of 64-bit signless integer values or memref of any type values -| `inputs` | tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `R` | tensor of 32-bit float values or tensor of 64-bit float values +| `T` | tensor of 64-bit signless integer values +| `inputs` | tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `outputs` | tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `outputs` | tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Mul` (::mlir::ONNXMulOp) @@ -4084,14 +4138,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `A` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values -| `B` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `A` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `B` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `C` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `C` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.Multinomial` (::mlir::ONNXMultinomialOp) @@ -4116,13 +4170,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 memref of any type values +| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `output` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values +| `output` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values ### `onnx.Neg` (::mlir::ONNXNegOp) @@ -4140,13 +4194,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 32-bit float values or tensor of 32-bit signless integer values or tensor of 8-bit signless integer values or tensor of 16-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `X` | tensor of 32-bit float values or tensor of 32-bit signless integer values or tensor of 8-bit signless integer values or tensor of 16-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 32-bit float values or tensor of 32-bit signless integer values or tensor of 8-bit signless integer values or tensor of 16-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `Y` | tensor of 32-bit float values or tensor of 32-bit signless integer values or tensor of 8-bit signless integer values or tensor of 16-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.NegativeLogLikelihoodLoss` (::mlir::ONNXNegativeLogLikelihoodLossOp) @@ -4254,15 +4308,15 @@ 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 memref of any type values -| `target` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values -| `weight` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type +| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `target` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values +| `weight` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type #### Results: | Result | Description | | :----: | ----------- | -| `loss` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `loss` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.NonMaxSuppression` (::mlir::ONNXNonMaxSuppressionOp) @@ -4290,17 +4344,17 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `boxes` | tensor of 32-bit float values or memref of any type values -| `scores` | tensor of 32-bit float values or memref of any type values -| `max_output_boxes_per_class` | tensor of 64-bit signless integer values or memref of any type values or none type -| `iou_threshold` | tensor of 32-bit float values or memref of any type values or none type -| `score_threshold` | tensor of 32-bit float values or memref of any type values or none type +| `boxes` | tensor of 32-bit float values +| `scores` | tensor of 32-bit float values +| `max_output_boxes_per_class` | tensor of 64-bit signless integer values or none type +| `iou_threshold` | tensor of 32-bit float values or none type +| `score_threshold` | tensor of 32-bit float values or none type #### Results: | Result | Description | | :----: | ----------- | -| `selected_indices` | tensor of 64-bit signless integer values or memref of any type values +| `selected_indices` | tensor of 64-bit signless integer values ### `onnx.NonZero` (::mlir::ONNXNonZeroOp) @@ -4319,13 +4373,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `X` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 64-bit signless integer values or memref of any type values +| `Y` | tensor of 64-bit signless integer values ### `onnx.NoValue` (::mlir::ONNXNoneOp) @@ -4384,13 +4438,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values or memref of any type values +| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 32-bit float values or memref of any type values +| `Y` | tensor of 32-bit float values ### `onnx.Not` (::mlir::ONNXNotOp) @@ -4406,13 +4460,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 1-bit signless integer values or memref of any type values +| `X` | tensor of 1-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 1-bit signless integer values or memref of any type values +| `Y` | tensor of 1-bit signless integer values ### `onnx.OneHotEncoder` (::mlir::ONNXOneHotEncoderOp) @@ -4443,13 +4497,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of string type values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `X` | tensor of string type values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 32-bit float values or memref of any type values +| `Y` | tensor of 32-bit float values ### `onnx.OneHot` (::mlir::ONNXOneHotOp) @@ -4489,15 +4543,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `indices` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `depth` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `values` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `indices` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `depth` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `values` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.Or` (::mlir::ONNXOrOp) @@ -4516,14 +4570,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `A` | tensor of 1-bit signless integer values or memref of any type values -| `B` | tensor of 1-bit signless integer values or memref of any type values +| `A` | tensor of 1-bit signless integer values +| `B` | tensor of 1-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `C` | tensor of 1-bit signless integer values or memref of any type values +| `C` | tensor of 1-bit signless integer values ### `onnx.PRelu` (::mlir::ONNXPReluOp) @@ -4545,14 +4599,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values -| `slope` | tensor of bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values +| `X` | tensor of bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values +| `slope` | tensor of bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values +| `Y` | tensor of bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values ### `onnx.Pad` (::mlir::ONNXPadOp) @@ -4648,15 +4702,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `pads` | tensor of 64-bit signless integer values or memref of any type values -| `constant_value` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values or none type +| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `pads` | tensor of 64-bit signless integer values +| `constant_value` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or none type #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.PadV11` (::mlir::ONNXPadV11Op) @@ -4752,15 +4806,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `pads` | tensor of 64-bit signless integer values or memref of any type values -| `constant_value` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type +| `data` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `pads` | tensor of 64-bit signless integer values +| `constant_value` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.PadV2` (::mlir::ONNXPadV2Op) @@ -4799,13 +4853,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `data` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Pow` (::mlir::ONNXPowOp) @@ -4824,14 +4878,32 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values -| `Y` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `X` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `Y` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `Z` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `Z` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values + +### `onnx.PrintSignature` (::mlir::ONNXPrintSignatureOp) + +ONNX Op to print type signature of its input operands + +"Print type signature of the op's input operands." + +#### Attributes: + +| Attribute | MLIR Type | Description | +| :-------: | :-------: | ----------- | +| `op_name` | ::mlir::StringAttr | string attribute + +#### Operands: + +| Operand | Description | +| :-----: | ----------- | +| `input` | tensor of any type values ### `onnx.QLinearConv` (::mlir::ONNXQLinearConvOp) @@ -4864,21 +4936,21 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `x` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values -| `x_scale` | tensor of 32-bit float values or memref of any type values -| `x_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values -| `w` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values -| `w_scale` | tensor of 32-bit float values or memref of any type values -| `w_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values -| `y_scale` | tensor of 32-bit float values or memref of any type values -| `y_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values -| `B` | tensor of 32-bit signless integer values or memref of any type values or none type +| `x` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values +| `x_scale` | tensor of 32-bit float values +| `x_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values +| `w` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values +| `w_scale` | tensor of 32-bit float values +| `w_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values +| `y_scale` | tensor of 32-bit float values +| `y_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values +| `B` | tensor of 32-bit signless integer values or none type #### Results: | Result | Description | | :----: | ----------- | -| `y` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values +| `y` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values ### `onnx.QLinearMatMul` (::mlir::ONNXQLinearMatMulOp) @@ -4904,20 +4976,20 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `a` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values -| `a_scale` | tensor of 32-bit float values or memref of any type values -| `a_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values -| `b` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values -| `b_scale` | tensor of 32-bit float values or memref of any type values -| `b_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values -| `y_scale` | tensor of 32-bit float values or memref of any type values -| `y_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values +| `a` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values +| `a_scale` | tensor of 32-bit float values +| `a_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values +| `b` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values +| `b_scale` | tensor of 32-bit float values +| `b_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values +| `y_scale` | tensor of 32-bit float values +| `y_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values #### Results: | Result | Description | | :----: | ----------- | -| `y` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values +| `y` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values ### `onnx.QuantizeLinear` (::mlir::ONNXQuantizeLinearOp) @@ -4943,15 +5015,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `x` | tensor of 32-bit float values or tensor of 32-bit signless integer values or memref of any type values -| `y_scale` | tensor of 32-bit float values or memref of any type values -| `y_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values or none type +| `x` | tensor of 32-bit float values or tensor of 32-bit signless integer values +| `y_scale` | tensor of 32-bit float values +| `y_zero_point` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or none type #### Results: | Result | Description | | :----: | ----------- | -| `y` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values or memref of any type values +| `y` | tensor of 8-bit signless integer values or tensor of 8-bit unsigned integer values ### `onnx.RNN` (::mlir::ONNXRNNOp) @@ -5039,19 +5111,19 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `W` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `R` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `B` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type -| `sequence_lens` | tensor of 32-bit signless integer values or memref of any type values or none type -| `initial_h` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `W` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `R` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `B` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type +| `sequence_lens` | tensor of 32-bit signless integer values or none type +| `initial_h` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type -| `Y_h` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type +| `Y_h` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type ### `onnx.RandomNormalLike` (::mlir::ONNXRandomNormalLikeOp) @@ -5082,13 +5154,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.RandomNormal` (::mlir::ONNXRandomNormalOp) @@ -5120,7 +5192,7 @@ Effects: MemoryEffects::Effect{} | Result | Description | | :----: | ----------- | -| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.RandomUniformLike` (::mlir::ONNXRandomUniformLikeOp) @@ -5151,13 +5223,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.RandomUniform` (::mlir::ONNXRandomUniformOp) @@ -5188,7 +5260,7 @@ Effects: MemoryEffects::Effect{} | Result | Description | | :----: | ----------- | -| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Range` (::mlir::ONNXRangeOp) @@ -5228,15 +5300,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `start` | tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values -| `limit` | tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values -| `delta` | tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values +| `start` | tensor of 32-bit float values or tensor of 64-bit float 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 +| `limit` | tensor of 32-bit float values or tensor of 64-bit float 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 +| `delta` | tensor of 32-bit float values or tensor of 64-bit float 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 #### Results: | Result | Description | | :----: | ----------- | -| `output` | tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values +| `output` | tensor of 32-bit float values or tensor of 64-bit float 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 ### `onnx.Reciprocal` (::mlir::ONNXReciprocalOp) @@ -5254,13 +5326,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.ReduceL1` (::mlir::ONNXReduceL1Op) @@ -5288,13 +5360,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.ReduceL2` (::mlir::ONNXReduceL2Op) @@ -5322,13 +5394,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.ReduceLogSumExp` (::mlir::ONNXReduceLogSumExpOp) @@ -5356,13 +5428,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.ReduceLogSum` (::mlir::ONNXReduceLogSumOp) @@ -5390,13 +5462,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.ReduceMax` (::mlir::ONNXReduceMaxOp) @@ -5424,13 +5496,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or tensor of 8-bit unsigned integer values or tensor of 8-bit signless integer values or memref of any type values +| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or tensor of 8-bit unsigned integer values or tensor of 8-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or tensor of 8-bit unsigned integer values or tensor of 8-bit signless integer values or memref of any type values +| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or tensor of 8-bit unsigned integer values or tensor of 8-bit signless integer values ### `onnx.ReduceMean` (::mlir::ONNXReduceMeanOp) @@ -5458,13 +5530,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.ReduceMin` (::mlir::ONNXReduceMinOp) @@ -5492,13 +5564,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or tensor of 8-bit unsigned integer values or tensor of 8-bit signless integer values or memref of any type values +| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or tensor of 8-bit unsigned integer values or tensor of 8-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or tensor of 8-bit unsigned integer values or tensor of 8-bit signless integer values or memref of any type values +| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or tensor of 8-bit unsigned integer values or tensor of 8-bit signless integer values ### `onnx.ReduceProd` (::mlir::ONNXReduceProdOp) @@ -5526,13 +5598,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.ReduceSum` (::mlir::ONNXReduceSumOp) @@ -5560,14 +5632,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values -| `axes` | tensor of 64-bit signless integer values or memref of any type values or none type +| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `axes` | tensor of 64-bit signless integer values or none type #### Results: | Result | Description | | :----: | ----------- | -| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.ReduceSumSquare` (::mlir::ONNXReduceSumSquareOp) @@ -5595,13 +5667,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.ReduceSumV11` (::mlir::ONNXReduceSumV11Op) @@ -5629,13 +5701,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `data` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `reduced` | tensor of 32-bit unsigned integer values or tensor of 64-bit unsigned integer values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Relu` (::mlir::ONNXReluOp) @@ -5653,13 +5725,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 32-bit float values or tensor of 32-bit signless integer values or tensor of 8-bit signless integer values or tensor of 16-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `X` | tensor of 32-bit float values or tensor of 32-bit signless integer values or tensor of 8-bit signless integer values or tensor of 16-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 32-bit float values or tensor of 32-bit signless integer values or tensor of 8-bit signless integer values or tensor of 16-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `Y` | tensor of 32-bit float values or tensor of 32-bit signless integer values or tensor of 8-bit signless integer values or tensor of 16-bit signless integer values or tensor of 64-bit signless integer values or tensor of 16-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.Reshape` (::mlir::ONNXReshapeOp) @@ -5670,25 +5742,33 @@ ONNX Reshape operation "At most one dimension of the new shape can be -1. In this case, the value is" "inferred from the size of the tensor and the remaining dimensions. A dimension" "could also be 0, in which case the actual dimension value is unchanged (i.e. taken" -"from the input tensor). Shape (second input) could be an empty shape, which means converting to a scalar." +"from the input tensor). If 'allowzero' is set, and the new shape includes 0, the" +"dimension will be set explicitly to zero (i.e. not taken from input tensor)." +"Shape (second input) could be an empty shape, which means converting to a scalar." "The input tensor's shape and the output tensor's shape are required to have the same number of elements." Interfaces: NoSideEffect (MemoryEffectOpInterface), ShapeInference Effects: MemoryEffects::Effect{} +#### Attributes: + +| Attribute | MLIR Type | Description | +| :-------: | :-------: | ----------- | +| `allowzero` | ::mlir::IntegerAttr | 64-bit signed integer attribute + #### Operands: | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `shape` | tensor of 64-bit signless integer values or memref of any type values +| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `shape` | tensor of 64-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `reshaped` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `reshaped` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.Resize` (::mlir::ONNXResizeOp) @@ -5717,16 +5797,16 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `roi` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values or none type -| `scales` | tensor of 32-bit float values or memref of any type values or none type -| `sizes` | tensor of 64-bit signless integer values or memref of any type values or none type +| `X` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `roi` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or none type +| `scales` | tensor of 32-bit float values or none type +| `sizes` | tensor of 64-bit signless integer values or none type #### Results: | Result | Description | | :----: | ----------- | -| `Y` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `Y` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.ResizeV10` (::mlir::ONNXResizeV10Op) @@ -5750,14 +5830,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `scales` | tensor of 32-bit float values or memref of any type values +| `X` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `scales` | tensor of 32-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `Y` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.ResizeV11` (::mlir::ONNXResizeV11Op) @@ -5786,16 +5866,16 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `roi` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `scales` | tensor of 32-bit float values or memref of any type values -| `sizes` | tensor of 64-bit signless integer values or memref of any type values or none type +| `X` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `roi` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `scales` | tensor of 32-bit float values +| `sizes` | tensor of 64-bit signless integer values or none type #### Results: | Result | Description | | :----: | ----------- | -| `Y` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `Y` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.Return` (::mlir::ONNXReturnOp) @@ -5877,14 +5957,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `sequence_lens` | tensor of 64-bit signless integer values or memref of any type values +| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `sequence_lens` | tensor of 64-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `Y` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.RoiAlign` (::mlir::ONNXRoiAlignOp) @@ -5920,15 +6000,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `rois` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `batch_indices` | tensor of 64-bit signless integer values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `rois` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `batch_indices` | tensor of 64-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Round` (::mlir::ONNXRoundOp) @@ -5956,13 +6036,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.SVMClassifier` (::mlir::ONNXSVMClassifierOp) @@ -5994,14 +6074,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values or memref of any type values +| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of string type values or tensor of 64-bit signless integer values or memref of any type values -| `Z` | tensor of 32-bit float values or memref of any type values +| `Y` | tensor of string type values or tensor of 64-bit signless integer values +| `Z` | tensor of 32-bit float values ### `onnx.SVMRegressor` (::mlir::ONNXSVMRegressorOp) @@ -6030,13 +6110,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values or memref of any type values +| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 32-bit float values or memref of any type values +| `Y` | tensor of 32-bit float values ### `onnx.Scaler` (::mlir::ONNXScalerOp) @@ -6059,13 +6139,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values or memref of any type values +| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 32-bit float values or memref of any type values +| `Y` | tensor of 32-bit float values ### `onnx.Scan` (::mlir::ONNXScanOp) @@ -6211,13 +6291,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `initial_state_and_scan_inputs` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `initial_state_and_scan_inputs` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `final_state_and_scan_outputs` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `final_state_and_scan_outputs` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.ScatterElements` (::mlir::ONNXScatterElementsOp) @@ -6289,15 +6369,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `indices` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values -| `updates` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `indices` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values +| `updates` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.ScatterND` (::mlir::ONNXScatterNDOp) @@ -6392,15 +6472,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `indices` | tensor of 64-bit signless integer values or memref of any type values -| `updates` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `indices` | tensor of 64-bit signless integer values +| `updates` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.Scatter` (::mlir::ONNXScatterOp) @@ -6474,15 +6554,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `indices` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values -| `updates` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `data` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `indices` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values +| `updates` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.Selu` (::mlir::ONNXSeluOp) @@ -6508,13 +6588,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.SequenceAt` (::mlir::ONNXSequenceAtOp) @@ -6532,14 +6612,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input_sequence` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or memref of any type values -| `position` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values +| `input_sequence` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values +| `position` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `tensor` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `tensor` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.SequenceConstruct` (::mlir::ONNXSequenceConstructOp) @@ -6556,13 +6636,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `inputs` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `inputs` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `output_sequence` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or memref of any type values +| `output_sequence` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values ### `onnx.SequenceEmpty` (::mlir::ONNXSequenceEmptyOp) @@ -6584,7 +6664,7 @@ Effects: MemoryEffects::Effect{} | Result | Description | | :----: | ----------- | -| `output` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or memref of any type values +| `output` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values ### `onnx.SequenceErase` (::mlir::ONNXSequenceEraseOp) @@ -6603,14 +6683,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input_sequence` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or memref of any type values -| `position` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values or none type +| `input_sequence` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values +| `position` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or none type #### Results: | Result | Description | | :----: | ----------- | -| `output_sequence` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or memref of any type values +| `output_sequence` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values ### `onnx.SequenceInsert` (::mlir::ONNXSequenceInsertOp) @@ -6630,15 +6710,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input_sequence` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or memref of any type values -| `tensor` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `position` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values or none type +| `input_sequence` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values +| `tensor` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `position` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or none type #### Results: | Result | Description | | :----: | ----------- | -| `output_sequence` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or memref of any type values +| `output_sequence` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values ### `onnx.SequenceLength` (::mlir::ONNXSequenceLengthOp) @@ -6654,13 +6734,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input_sequence` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or memref of any type values +| `input_sequence` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values #### Results: | Result | Description | | :----: | ----------- | -| `length` | tensor of 64-bit signless integer values or memref of any type values +| `length` | tensor of 64-bit signless integer values ### `onnx.Shape` (::mlir::ONNXShapeOp) @@ -6676,13 +6756,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `shape` | tensor of 64-bit signless integer values or memref of any type values +| `shape` | tensor of 64-bit signless integer values ### `onnx.Shrink` (::mlir::ONNXShrinkOp) @@ -6708,13 +6788,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Sigmoid` (::mlir::ONNXSigmoidOp) @@ -6732,13 +6812,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.Sign` (::mlir::ONNXSignOp) @@ -6755,13 +6835,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `output` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.Sin` (::mlir::ONNXSinOp) @@ -6777,13 +6857,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 memref of any type values +| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Sinh` (::mlir::ONNXSinhOp) @@ -6799,13 +6879,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 memref of any type values +| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Size` (::mlir::ONNXSizeOp) @@ -6821,13 +6901,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `size` | tensor of 64-bit signless integer values or memref of any type values +| `size` | tensor of 64-bit signless integer values ### `onnx.Slice` (::mlir::ONNXSliceOp) @@ -6878,17 +6958,17 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `starts` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values -| `ends` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values -| `axes` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values or none type -| `steps` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values or none type +| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `starts` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values +| `ends` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values +| `axes` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or none type +| `steps` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or none type #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.SoftmaxCrossEntropyLoss` (::mlir::ONNXSoftmaxCrossEntropyLossOp) @@ -6941,16 +7021,16 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `scores` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values -| `labels` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values -| `weights` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values or none type +| `scores` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `labels` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values +| `weights` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or none type #### 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 bfloat16 type values or memref of any type values -| `log_prob` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values or none type +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `log_prob` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or none type ### `onnx.Softmax` (::mlir::ONNXSoftmaxOp) @@ -6978,13 +7058,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 bfloat16 type values or memref of any 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 bfloat16 type 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 bfloat16 type values or memref of any 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 bfloat16 type values ### `onnx.Softplus` (::mlir::ONNXSoftplusOp) @@ -7002,13 +7082,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Softsign` (::mlir::ONNXSoftsignOp) @@ -7024,13 +7104,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 memref of any type values +| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.SpaceToDepth` (::mlir::ONNXSpaceToDepthOp) @@ -7054,13 +7134,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `input` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.Split` (::mlir::ONNXSplitOp) @@ -7084,14 +7164,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `split` | tensor of 64-bit signless integer values or memref of any type values or none type +| `input` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `split` | tensor of 64-bit signless integer values or none type #### Results: | Result | Description | | :----: | ----------- | -| `outputs` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `outputs` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.SplitToSequence` (::mlir::ONNXSplitToSequenceOp) @@ -7123,14 +7203,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `split` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values or none type +| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `split` | tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or none type #### Results: | Result | Description | | :----: | ----------- | -| `output_sequence` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values or memref of any type values +| `output_sequence` | SeqType of tensor of 8-bit unsigned integer values values or SeqType of tensor of 16-bit unsigned integer values values or SeqType of tensor of 32-bit unsigned integer values values or SeqType of tensor of 64-bit unsigned integer values values or SeqType of tensor of 8-bit signless integer values values or SeqType of tensor of 16-bit signless integer values values or SeqType of tensor of 32-bit signless integer values values or SeqType of tensor of 64-bit signless integer values values or SeqType of tensor of 16-bit float values values or SeqType of tensor of 32-bit float values values or SeqType of tensor of 64-bit float values values or SeqType of tensor of string type values values or SeqType of tensor of 1-bit signless integer values values or SeqType of tensor of complex type with 32-bit float elements values values or SeqType of tensor of complex type with 64-bit float elements values values ### `onnx.SplitV11` (::mlir::ONNXSplitV11Op) @@ -7155,13 +7235,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `input` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `outputs` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `outputs` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.Sqrt` (::mlir::ONNXSqrtOp) @@ -7179,13 +7259,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.Squeeze` (::mlir::ONNXSqueezeOp) @@ -7204,14 +7284,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `axes` | tensor of 64-bit signless integer values or memref of any type values or none type +| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `axes` | tensor of 64-bit signless integer values or none type #### Results: | Result | Description | | :----: | ----------- | -| `squeezed` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `squeezed` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.SqueezeV11` (::mlir::ONNXSqueezeV11Op) @@ -7236,13 +7316,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `data` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `squeezed` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `squeezed` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.StringNormalizer` (::mlir::ONNXStringNormalizerOp) @@ -7275,13 +7355,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of string type values or memref of any type values +| `X` | tensor of string type values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of string type values or memref of any type values +| `Y` | tensor of string type values ### `onnx.Sub` (::mlir::ONNXSubOp) @@ -7301,14 +7381,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `A` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values -| `B` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `A` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values +| `B` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `C` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `C` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.Sum` (::mlir::ONNXSumOp) @@ -7326,13 +7406,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data_0` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `data_0` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values #### Results: | Result | Description | | :----: | ----------- | -| `sum` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values or memref of any type values +| `sum` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of bfloat16 type values ### `onnx.Tan` (::mlir::ONNXTanOp) @@ -7348,13 +7428,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 memref of any type values +| `input` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float 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 memref of any type values +| `output` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Tanh` (::mlir::ONNXTanhOp) @@ -7370,13 +7450,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 bfloat16 type values or memref of any 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 bfloat16 type 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 bfloat16 type values or memref of any 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 bfloat16 type values ### `onnx.TfIdfVectorizer` (::mlir::ONNXTfIdfVectorizerOp) @@ -7432,13 +7512,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of string type values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values or memref of any type values +| `X` | tensor of string type values or tensor of 32-bit signless integer values or tensor of 64-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 32-bit float values or memref of any type values +| `Y` | tensor of 32-bit float values ### `onnx.ThresholdedRelu` (::mlir::ONNXThresholdedReluOp) @@ -7462,13 +7542,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `X` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values +| `Y` | tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values ### `onnx.Tile` (::mlir::ONNXTileOp) @@ -7486,14 +7566,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `input` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `repeats` | tensor of 64-bit signless integer values or memref of any type values +| `input` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `repeats` | tensor of 64-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.TopK` (::mlir::ONNXTopKOp) @@ -7530,15 +7610,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `K` | tensor of 64-bit signless integer values or memref of any type values +| `X` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `K` | tensor of 64-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `Values` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or memref of any type values -| `Indices` | tensor of 64-bit signless integer values or memref of any type values +| `Values` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values +| `Indices` | tensor of 64-bit signless integer values ### `onnx.Transpose` (::mlir::ONNXTransposeOp) @@ -7562,13 +7642,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `transposed` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `transposed` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.TreeEnsembleClassifier` (::mlir::ONNXTreeEnsembleClassifierOp) @@ -7614,14 +7694,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values or memref of any type values +| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of string type values or tensor of 64-bit signless integer values or memref of any type values -| `Z` | tensor of 32-bit float values or memref of any type values +| `Y` | tensor of string type values or tensor of 64-bit signless integer values +| `Z` | tensor of 32-bit float values ### `onnx.TreeEnsembleRegressor` (::mlir::ONNXTreeEnsembleRegressorOp) @@ -7668,13 +7748,53 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values or memref of any type values +| `X` | tensor of 32-bit float values or tensor of 64-bit float values or tensor of 64-bit signless integer values or tensor of 32-bit signless integer values + +#### Results: + +| Result | Description | +| :----: | ----------- | +| `Y` | tensor of 32-bit float values + +### `onnx.Trilu` (::mlir::ONNXTriluOp) + +ONNX Trilu operation + +"Given a 2-D matrix or batches of 2-D matrices, returns the upper or lower triangular part of the tensor(s)." +"The attribute \"upper\" determines whether the upper or lower part is retained. If set to true," +"the upper triangular matrix is retained. Lower triangular matrix is retained otherwise." +"Default value for the \"upper\" attribute is true." +"Trilu takes one input tensor of shape [*, N, M], where * is zero or more batch dimensions. The upper triangular part consists" +"of the elements on and above the given diagonal (k). The lower triangular part consists of elements on and below the diagonal." +"All other elements in the matrix are set to zero." +"If k = 0, the triangular part on and above/below the main diagonal is retained." +"If upper is set to true, a positive k retains the upper triangular matrix excluding the main diagonal and (k-1) diagonals above it." +"A negative k value retains the main diagonal and |k| diagonals below it." +"If upper is set to false, a positive k retains the lower triangular matrix including the main diagonal and k diagonals above it." +"A negative k value excludes the main diagonal and (|k|-1) diagonals below it." + +Interfaces: NoSideEffect (MemoryEffectOpInterface), ShapeInference + +Effects: MemoryEffects::Effect{} + +#### Attributes: + +| Attribute | MLIR Type | Description | +| :-------: | :-------: | ----------- | +| `upper` | ::mlir::IntegerAttr | 64-bit signed integer attribute + +#### Operands: + +| Operand | Description | +| :-----: | ----------- | +| `input` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `k` | tensor of 64-bit signless integer values or none type #### Results: | Result | Description | | :----: | ----------- | -| `Y` | tensor of 32-bit float values or memref of any type values +| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.Unique` (::mlir::ONNXUniqueOp) @@ -7771,16 +7891,16 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `X` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `indices` | tensor of 64-bit signless integer values or memref of any type values or none type -| `inverse_indices` | tensor of 64-bit signless integer values or memref of any type values or none type -| `counts` | tensor of 64-bit signless integer values or memref of any type values or none type +| `Y` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `indices` | tensor of 64-bit signless integer values or none type +| `inverse_indices` | tensor of 64-bit signless integer values or none type +| `counts` | tensor of 64-bit signless integer values or none type ### `onnx.Unsqueeze` (::mlir::ONNXUnsqueezeOp) @@ -7807,14 +7927,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `axes` | tensor of 64-bit signless integer values or memref of any type values +| `data` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `axes` | tensor of 64-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `expanded` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `expanded` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.UnsqueezeV11` (::mlir::ONNXUnsqueezeV11Op) @@ -7847,13 +7967,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `data` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `data` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `expanded` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `expanded` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.Upsample` (::mlir::ONNXUpsampleOp) @@ -7877,14 +7997,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `scales` | tensor of 32-bit float values or memref of any type values +| `X` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `scales` | tensor of 32-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `Y` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.UpsampleV7` (::mlir::ONNXUpsampleV7Op) @@ -7909,13 +8029,13 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `X` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `Y` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.UpsampleV9` (::mlir::ONNXUpsampleV9Op) @@ -7939,14 +8059,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `scales` | tensor of 32-bit float values or memref of any type values +| `X` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `scales` | tensor of 32-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Y` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `Y` | 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 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 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.Where` (::mlir::ONNXWhereOp) @@ -7970,15 +8090,15 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `condition` | tensor of 1-bit signless integer values or memref of any type values -| `X` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values -| `Y` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `condition` | tensor of 1-bit signless integer values +| `X` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values +| `Y` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values #### Results: | Result | Description | | :----: | ----------- | -| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values or memref of any type values +| `output` | 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 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 bfloat16 type values or tensor of 16-bit float values or tensor of 32-bit float values or tensor of 64-bit float values or tensor of string type values or tensor of 1-bit signless integer values or tensor of complex type with 32-bit float elements values or tensor of complex type with 64-bit float elements values ### `onnx.Xor` (::mlir::ONNXXorOp) @@ -7997,14 +8117,14 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `A` | tensor of 1-bit signless integer values or memref of any type values -| `B` | tensor of 1-bit signless integer values or memref of any type values +| `A` | tensor of 1-bit signless integer values +| `B` | tensor of 1-bit signless integer values #### Results: | Result | Description | | :----: | ----------- | -| `C` | tensor of 1-bit signless integer values or memref of any type values +| `C` | tensor of 1-bit signless integer values ### `onnx.ZipMap` (::mlir::ONNXZipMapOp) @@ -8030,11 +8150,11 @@ Effects: MemoryEffects::Effect{} | Operand | Description | | :-----: | ----------- | -| `X` | tensor of 32-bit float values or memref of any type values +| `X` | tensor of 32-bit float values #### Results: | Result | Description | | :----: | ----------- | -| `Z` | SeqType of tuple with any combination of string type or 32-bit float values values or SeqType of tuple with any combination of 64-bit signless integer or 32-bit float values values or memref of any type values +| `Z` | SeqType of tuple with any combination of string type or 32-bit float values values or SeqType of tuple with any combination of 64-bit signless integer or 32-bit float values values diff --git a/docs/Dialects/zhigh.md b/docs/Dialects/zhigh.md index 28a5fee02e8..faaf6d4ff66 100644 --- a/docs/Dialects/zhigh.md +++ b/docs/Dialects/zhigh.md @@ -79,6 +79,37 @@ Effects: MemoryEffects::Effect{} | :----: | ----------- | | `output` | unranked.tensor of 32-bit float values or 4D tensor of 32-bit float values with layout NHWC +### `zhigh.Concat` (::onnx_mlir::zhigh::ZHighConcatOp) + +ZHigh Concat operation to concatenate stickified tensors + +"Concatenate a list of tensors into a single tensor. All input tensors must" +"have the same shape, except for the dimension size of the axis" +"to concatenate on. Users must ensure that it is safe to concatenate" +"stickified tensors for the given axis." + +Interfaces: NoSideEffect (MemoryEffectOpInterface), ShapeInference + +Effects: MemoryEffects::Effect{} + +#### Attributes: + +| Attribute | MLIR Type | Description | +| :-------: | :-------: | ----------- | +| `axis` | ::mlir::IntegerAttr | 64-bit signed integer attribute + +#### Operands: + +| Operand | Description | +| :-----: | ----------- | +| `inputs` | unranked.tensor of 32-bit float values or 4D tensor of 32-bit float values with layout _4D or unranked.tensor of 32-bit float values or 4D tensor of 32-bit float values with layout NHWC or memref of any type values + +#### Results: + +| Result | Description | +| :----: | ----------- | +| `concat_result` | unranked.tensor of 32-bit float values or 4D tensor of 32-bit float values with layout _4D or unranked.tensor of 32-bit float values or 4D tensor of 32-bit float values with layout NHWC or memref of any type values + ### `zhigh.Conv2D` (::onnx_mlir::zhigh::ZHighConv2DOp) ZHigh 2D convolution operation diff --git a/docs/Dialects/zlow.md b/docs/Dialects/zlow.md index 98c0f1a85de..3864a327324 100644 --- a/docs/Dialects/zlow.md +++ b/docs/Dialects/zlow.md @@ -141,6 +141,27 @@ Traits: MemRefsNormalizable | `shape` | memref of 64-bit signless integer values | `Out` | memref of any type values +### `zlow.dummy` (::onnx_mlir::zlow::ZLowDummyOp) + +ZLow dummy operation that behaves like identity + +"ZLow operation to forward the input value to the output value." +"It will be removed if canonicalization is called." + +Traits: MemRefsNormalizable + +#### Operands: + +| Operand | Description | +| :-----: | ----------- | +| `input` | memref of any type values + +#### Results: + +| Result | Description | +| :----: | ----------- | +| `output` | memref of any type values + ### `zlow.exp` (::onnx_mlir::zlow::ZLowExpOp) ZLow exp operation @@ -177,6 +198,7 @@ ZLow gru operation " - 5th item: hiddenSize" "direction accepts "forward", "reverse", or "bidirectional" "return_all_steps: -1 returns all timesteps, 0: returns only the last timestep. +"prev_layer for where input comes is "none", "uni", or "bidir" Traits: MemRefsNormalizable @@ -186,6 +208,7 @@ Traits: MemRefsNormalizable | :-------: | :-------: | ----------- | | `direction` | ::mlir::StringAttr | string attribute | `return_all_steps` | ::mlir::IntegerAttr | 64-bit signed integer attribute +| `prev_layer` | ::mlir::StringAttr | string attribute #### Operands: @@ -215,6 +238,7 @@ ZLow lstm operation " - 5th item: hiddenSize" "direction accepts "forward", "reverse", or "bidirectional" "return_all_steps: -1 returns all timesteps, 0: returns only the last timestep. +"prev_layer for where input comes is "none", "uni", or "bidir" Traits: MemRefsNormalizable @@ -224,6 +248,7 @@ Traits: MemRefsNormalizable | :-------: | :-------: | ----------- | | `direction` | ::mlir::StringAttr | string attribute | `return_all_steps` | ::mlir::IntegerAttr | 64-bit signed integer attribute +| `prev_layer` | ::mlir::StringAttr | string attribute #### Operands: @@ -505,6 +530,12 @@ ZLow stick operation for GRU Traits: MemRefsNormalizable +#### Attributes: + +| Attribute | MLIR Type | Description | +| :-------: | :-------: | ----------- | +| `prev_layer` | ::mlir::StringAttr | string attribute + #### Operands: | Operand | Description | @@ -524,6 +555,12 @@ ZLow stick operation for LSTM Traits: MemRefsNormalizable +#### Attributes: + +| Attribute | MLIR Type | Description | +| :-------: | :-------: | ----------- | +| `prev_layer` | ::mlir::StringAttr | string attribute + #### Operands: | Operand | Description | diff --git a/src/Accelerators/NNPA/Compiler/NNPACompilerUtils.cpp b/src/Accelerators/NNPA/Compiler/NNPACompilerUtils.cpp index 2f3ca0a5b5e..e14331a6b78 100644 --- a/src/Accelerators/NNPA/Compiler/NNPACompilerUtils.cpp +++ b/src/Accelerators/NNPA/Compiler/NNPACompilerUtils.cpp @@ -138,7 +138,8 @@ void addPassesNNPA(mlir::OwningOpRef &module, optLevel = OptLevel::O2; else if (optStr == "-O3") optLevel = OptLevel::O3; - addONNXToKrnlPasses(pm, optLevel); + addONNXToKrnlPasses( + pm, optLevel, /*enableCSE*/ true, instrumentONNXSignature); if (nnpaEmissionTarget >= EmitZLowIR) emissionTarget = EmitMLIR; diff --git a/src/Compiler/CompilerOptions.cpp b/src/Compiler/CompilerOptions.cpp index c4c4c2a126c..04139e5313f 100644 --- a/src/Compiler/CompilerOptions.cpp +++ b/src/Compiler/CompilerOptions.cpp @@ -147,6 +147,10 @@ llvm::cl::bits instrumentControlBits( InstrumentReportMemory, "instrument runtime reports memory usage")), llvm::cl::cat(OnnxMlirOptions)); +llvm::cl::opt instrumentONNXSignature("instrument-onnx-signature", + llvm::cl::desc("Instrument ONNX ops to print the type of their inputs"), + llvm::cl::init(false), llvm::cl::cat(OnnxMlirOptions)); + llvm::cl::opt enableMemoryBundling("enable-memory-bundling", llvm::cl::desc( "Enable memory bundling related optimizations (default=false)\n" diff --git a/src/Compiler/CompilerOptions.hpp b/src/Compiler/CompilerOptions.hpp index df49e658a23..ff0d1546f07 100644 --- a/src/Compiler/CompilerOptions.hpp +++ b/src/Compiler/CompilerOptions.hpp @@ -50,6 +50,7 @@ extern llvm::cl::opt verifyInputTensors; extern llvm::cl::opt instrumentONNXOps; extern llvm::cl::bits instrumentControlBits; +extern llvm::cl::opt instrumentONNXSignature; extern llvm::cl::opt enableMemoryBundling; extern llvm::cl::opt onnxOpTransformThreshold; extern llvm::cl::opt onnxOpTransformReport; diff --git a/src/Compiler/CompilerPasses.cpp b/src/Compiler/CompilerPasses.cpp index 5ad5782e5bd..466dbb4c6a1 100644 --- a/src/Compiler/CompilerPasses.cpp +++ b/src/Compiler/CompilerPasses.cpp @@ -69,7 +69,8 @@ void addONNXToMLIRPasses(mlir::PassManager &pm) { pm.addPass(mlir::createSymbolDCEPass()); } -void addONNXToKrnlPasses(mlir::PassManager &pm, int optLevel, bool enableCSE) { +void addONNXToKrnlPasses(mlir::PassManager &pm, int optLevel, bool enableCSE, + bool enableInstrumentONNXSignature) { if (enableCSE) // Eliminate common sub-expressions before lowering to Krnl. // TODO: enable this by default when we make sure it works flawlessly. @@ -79,6 +80,9 @@ void addONNXToKrnlPasses(mlir::PassManager &pm, int optLevel, bool enableCSE) { // Add instrumentation for Onnx Ops pm.addNestedPass(onnx_mlir::createInstrumentONNXPass( instrumentONNXOps, instrumentControlBits.getBits())); + if (enableInstrumentONNXSignature) + pm.addNestedPass( + onnx_mlir::createInstrumentONNXSignaturePass()); pm.addPass(onnx_mlir::createLowerToKrnlPass(optLevel)); // An additional pass of canonicalization is helpful because lowering // from ONNX dialect to Standard dialect exposes additional canonicalization @@ -162,7 +166,8 @@ void addPasses(mlir::OwningOpRef &module, mlir::PassManager &pm, if (emissionTarget >= EmitMLIR) { if (inputIRLevel <= ONNXLevel) - addONNXToKrnlPasses(pm, OptimizationLevel); + addONNXToKrnlPasses( + pm, OptimizationLevel, /*enableCSE*/ true, instrumentONNXSignature); if (inputIRLevel <= MLIRLevel) addKrnlToAffinePasses(pm); } diff --git a/src/Compiler/CompilerPasses.hpp b/src/Compiler/CompilerPasses.hpp index 09bcbf15202..ea4e8587b5b 100644 --- a/src/Compiler/CompilerPasses.hpp +++ b/src/Compiler/CompilerPasses.hpp @@ -17,8 +17,8 @@ namespace onnx_mlir { void addONNXToMLIRPasses(mlir::PassManager &pm); -void addONNXToKrnlPasses( - mlir::PassManager &pm, int optLevel, bool enableCSE = true); +void addONNXToKrnlPasses(mlir::PassManager &pm, int optLevel, bool enableCSE, + bool enableInstrumentONNXSignature); void addKrnlToAffinePasses(mlir::PassManager &pm); void addKrnlToLLVMPasses( mlir::OpPassManager &pm, bool enableCSE, bool verifyInputTensors); diff --git a/src/Conversion/ONNXToKrnl/CMakeLists.txt b/src/Conversion/ONNXToKrnl/CMakeLists.txt index 63d4d0d049c..0cd5dc8c498 100644 --- a/src/Conversion/ONNXToKrnl/CMakeLists.txt +++ b/src/Conversion/ONNXToKrnl/CMakeLists.txt @@ -34,9 +34,10 @@ add_onnx_mlir_library(OMONNXToKrnl Sequence/SequenceErase.cpp Sequence/SequenceInsert.cpp Sequence/SequenceLength.cpp + ConvertONNXToKrnl.cpp Tensor/ArgMax.cpp - Tensor/Concat.cpp Tensor/Compress.cpp + Tensor/Concat.cpp Tensor/Constant.cpp Tensor/ConstantOfShape.cpp Tensor/DepthToSpace.cpp @@ -49,22 +50,22 @@ add_onnx_mlir_library(OMONNXToKrnl Tensor/NonZero.cpp Tensor/OneHot.cpp Tensor/Pad.cpp + Tensor/PrintSignature.cpp + Tensor/Range.cpp + Tensor/Reshape.cpp + Tensor/Resize.cpp + Tensor/ReverseSequence.cpp Tensor/ScatterElements.cpp Tensor/ScatterND.cpp - Tensor/Squeeze.cpp - Tensor/Split.cpp Tensor/Shape.cpp Tensor/Size.cpp Tensor/Slice.cpp Tensor/SpaceToDepth.cpp - Tensor/Reshape.cpp + Tensor/Split.cpp + Tensor/Squeeze.cpp Tensor/Tile.cpp Tensor/Transpose.cpp Tensor/Unsqueeze.cpp - Tensor/Range.cpp - Tensor/Resize.cpp - Tensor/ReverseSequence.cpp - ConvertONNXToKrnl.cpp LINK_LIBS PUBLIC Accelerator diff --git a/src/Conversion/ONNXToKrnl/ConvertONNXToKrnl.cpp b/src/Conversion/ONNXToKrnl/ConvertONNXToKrnl.cpp index a6d93fb5d0c..e9fffa1f902 100644 --- a/src/Conversion/ONNXToKrnl/ConvertONNXToKrnl.cpp +++ b/src/Conversion/ONNXToKrnl/ConvertONNXToKrnl.cpp @@ -230,6 +230,7 @@ void populateONNXToKrnlConversionPattern(RewritePatternSet &patterns, populateLoweringONNXExpandOpPattern(patterns, typeConverter, ctx); populateLoweringONNXOneHotOpPattern(patterns, typeConverter, ctx); populateLoweringONNXCompressOpPattern(patterns, typeConverter, ctx); + populateLoweringONNXPrintSignaturePattern(patterns, typeConverter, ctx); // Neural network populateLoweringONNXConvOpPattern(patterns, typeConverter, ctx); populateLoweringONNXNormalizationOpPattern(patterns, typeConverter, ctx); diff --git a/src/Conversion/ONNXToKrnl/ONNXToKrnlCommon.hpp b/src/Conversion/ONNXToKrnl/ONNXToKrnlCommon.hpp index 0df5db18dc2..92d0101ec05 100644 --- a/src/Conversion/ONNXToKrnl/ONNXToKrnlCommon.hpp +++ b/src/Conversion/ONNXToKrnl/ONNXToKrnlCommon.hpp @@ -376,6 +376,8 @@ void populateLoweringONNXOneHotOpPattern( RewritePatternSet &, TypeConverter &, MLIRContext *); void populateLoweringONNXCompressOpPattern( RewritePatternSet &, TypeConverter &, MLIRContext *); +void populateLoweringONNXPrintSignaturePattern( + RewritePatternSet &, TypeConverter &, MLIRContext *); bool checkOpResultIsUsedByGetRef(memref::AllocOp *allocOp); diff --git a/src/Conversion/ONNXToKrnl/Tensor/PrintSignature.cpp b/src/Conversion/ONNXToKrnl/Tensor/PrintSignature.cpp new file mode 100644 index 00000000000..39f472f4199 --- /dev/null +++ b/src/Conversion/ONNXToKrnl/Tensor/PrintSignature.cpp @@ -0,0 +1,55 @@ +/* + * SPDX-License-Identifier: Apache-2.0 + */ + +//===---------------- Concat.cpp - Lowering Concat Op -------------------===// +// +// Copyright 2019-2022 The IBM Research Authors. +// +// ============================================================================= +// +// This file lowers the ONNX Print Signature Operator to Krnl dialect. +// +//===----------------------------------------------------------------------===// + +#include "src/Conversion/ONNXToKrnl/ONNXToKrnlCommon.hpp" +#include "src/Dialect/Krnl/KrnlHelper.hpp" +#include "src/Dialect/ONNX/ShapeInference/ONNXShapeHelper.hpp" + +using namespace mlir; + +namespace onnx_mlir { + +struct ONNXPrintSignatureLowering : public ConversionPattern { + ONNXPrintSignatureLowering(TypeConverter &typeConverter, MLIRContext *ctx) + : ConversionPattern( + typeConverter, ONNXPrintSignatureOp::getOperationName(), 1, ctx) {} + + LogicalResult matchAndRewrite(Operation *op, ArrayRef operands, + ConversionPatternRewriter &rewriter) const final { + // Gather info. + auto loc = op->getLoc(); + MultiDialectBuilder create(rewriter, loc); + ONNXPrintSignatureOp printSignatureOp = + llvm::dyn_cast(op); + ONNXPrintSignatureOpAdaptor operandAdaptor(operands); + + std::string opName(printSignatureOp.op_name().data()); + std::string msg = opName; + create.krnl.printf(msg); + for (Value oper : operandAdaptor.input()) { + msg = "%t "; + create.krnl.printTensor(msg, oper); + } + Value noneValue; + rewriter.replaceOpWithNewOp(op, "\n", noneValue); + return success(); + } +}; + +void populateLoweringONNXPrintSignaturePattern(RewritePatternSet &patterns, + TypeConverter &typeConverter, MLIRContext *ctx) { + patterns.insert(typeConverter, ctx); +} + +} // namespace onnx_mlir diff --git a/src/Dialect/Krnl/Krnl.td b/src/Dialect/Krnl/Krnl.td index 2325217f9bd..b41f359626d 100644 --- a/src/Dialect/Krnl/Krnl.td +++ b/src/Dialect/Krnl/Krnl.td @@ -1135,6 +1135,12 @@ def KrnlPrintTensorOp : Op let summary = "Print a tensor."; let description = [{ This operation can be used to generate a call to a runtime function which prints a tensor. + At the begining of the msg string, user can add formatting instructions. The flags are: + %s: detailed signature (including shape, type, offsets), + %t: compact type (ala MLIR: 32x16xfloat), + %d: data values. + When no formatting is provided, "%s%d" is used (detailed signature and data) by default. + Print operation ends with a newline, except when only requesting a compact types (%t). }]; let arguments = (ins StrAttr:$msg, AnyMemRef:$input); diff --git a/src/Dialect/Krnl/KrnlHelper.cpp b/src/Dialect/Krnl/KrnlHelper.cpp index 12f8e647f87..7a8a93b4ae2 100644 --- a/src/Dialect/Krnl/KrnlHelper.cpp +++ b/src/Dialect/Krnl/KrnlHelper.cpp @@ -33,7 +33,10 @@ ParseResult KrnlDialectOperandParser::ParseOptionalOperand( if (operandRefQueue.empty()) { // Parse operand types: llvm::SmallVector operand_refs; - parser.parseOperandList(operand_refs); + if (failed(parser.parseOperandList(operand_refs))) { + operand = nullptr; + return failure(); + } // Record operands: for (auto &operand_ref : operand_refs) @@ -46,7 +49,10 @@ ParseResult KrnlDialectOperandParser::ParseOptionalOperand( operandRefQueue.pop(); llvm::SmallVector operands; - parser.resolveOperand(operand_ref, operandType, operands); + if (parser.resolveOperand(operand_ref, operandType, operands)) { + operand = nullptr; + return failure(); + } operand = operands.front(); return success(); } else { diff --git a/src/Dialect/ONNX/AdditionalONNXOps.td b/src/Dialect/ONNX/AdditionalONNXOps.td index b5efc1ad902..b239e114f53 100644 --- a/src/Dialect/ONNX/AdditionalONNXOps.td +++ b/src/Dialect/ONNX/AdditionalONNXOps.td @@ -98,3 +98,12 @@ def ONNXCallOp : ONNX_Op<"ONNX_Call", } }]; } + +def ONNXPrintSignatureOp:ONNX_Op<"PrintSignature", []> { + let summary = "ONNX Op to print type signature of its input operands"; + let description = [{ + "Print type signature of the op's input operands." + }]; + + let arguments = (ins StrAttr:$op_name, Variadic>:$input); +} diff --git a/src/InitOMPasses.hpp b/src/InitOMPasses.hpp index a8a9c2886e8..60fcde5b207 100644 --- a/src/InitOMPasses.hpp +++ b/src/InitOMPasses.hpp @@ -42,6 +42,10 @@ void initOMPasses(int optLevel) { return createInstrumentONNXPass(); }); + mlir::registerPass([]() -> std::unique_ptr { + return createInstrumentONNXSignaturePass(); + }); + mlir::registerPass([]() -> std::unique_ptr { return createONNXPreKrnlVerifyPass(); }); diff --git a/src/Pass/Passes.hpp b/src/Pass/Passes.hpp index 93573e24135..f8f0a3e9bac 100644 --- a/src/Pass/Passes.hpp +++ b/src/Pass/Passes.hpp @@ -43,6 +43,10 @@ std::unique_ptr createInstrumentONNXPass(); std::unique_ptr createInstrumentONNXPass( llvm::StringRef ops, int actions); +/// Passes for instrumenting the ONNX ops to print their operand type +/// signatures at runtime. +std::unique_ptr createInstrumentONNXSignaturePass(); + /// Pass for verifying Onnx ops before lowering to Krnl std::unique_ptr createONNXPreKrnlVerifyPass(); diff --git a/src/Runtime/OMTensor.inc b/src/Runtime/OMTensor.inc index 6658d7a54f1..b72520a2e03 100644 --- a/src/Runtime/OMTensor.inc +++ b/src/Runtime/OMTensor.inc @@ -418,20 +418,54 @@ void omTensorPrint(const char *msg, const OMTensor *tensor) { const int64_t *strides = omTensorGetStrides(tensor); void *dataPtr = omTensorGetDataPtr(tensor); + // Process format if any (default is signature and data). + bool signatureInfo = false; + bool typeInfo = false; + bool dataInfo = false; + while (strlen(msg) >= 2 && msg[0] == '%') { + if (msg[1] == 's') // S for signature. + signatureInfo = true; + else if (msg[1] == 't') // T for type only + typeInfo = true; + else if (msg[1] == 'd') + dataInfo = true; + else { + printf( + "warning: unknown format `%c`, print signature and data\n", msg[1]); + signatureInfo = dataInfo = true; + } + // Skip the format when printing message. + msg += 2; + } + + // Print message. if (msg) printf("%s", msg); - printf("\trank = %lld\n", (long long)rank); - printf("\tdataType = %s\n", getDataTypeName(dataType)); - printf("\tnumElems = %lld\n", (long long)omTensorGetNumElems(tensor)); - printf("\tshape: "); - for (int64_t i = 0; i < rank; i++) - printf("[%lld]", (long long)shape[i]); - printf("\n"); - printf("\tstrides: "); - for (int64_t i = 0; i < rank; i++) - printf("[%lld]", (long long)strides[i]); - printf("\n"); + if (signatureInfo) { + // Print detailed info. + printf("\trank = %lld\n", (long long)rank); + printf("\tdataType = %s\n", getDataTypeName(dataType)); + printf("\tnumElems = %lld\n", (long long)omTensorGetNumElems(tensor)); + printf("\tshape: "); + for (int64_t i = 0; i < rank; i++) + printf("[%lld]", (long long)shape[i]); + printf("\n"); + printf("\tstrides: "); + for (int64_t i = 0; i < rank; i++) + printf("[%lld]", (long long)strides[i]); + printf("\n"); + } + if (typeInfo) { + // Print compact type like in MLIR, e.g. 32x16xf32. + if (signatureInfo) + printf("\ttype: "); + for (int64_t i = 0; i < rank; i++) + printf("%lldx", (long long)shape[i]); + printf("%s", getDataTypeName(dataType)); + if (signatureInfo) + printf("\n"); + } #define LOOP_1(INDEX, IV, UB) \ printf("["); \ @@ -474,28 +508,30 @@ void omTensorPrint(const char *msg, const OMTensor *tensor) { } \ printf("]"); - printf("\tdata: ("); - switch (rank) { - case 1: { - int64_t indexes[1]; - LOOP_1(0, i, shape[0]) - } break; - case 2: { - int64_t indexes[2]; - LOOP_2(0, i, shape[0], j, shape[1]) - } break; - case 3: { - int64_t indexes[3]; - LOOP_3(0, i, shape[0], j, shape[1], k, shape[2]) - } break; - case 4: { - int64_t indexes[4]; - LOOP_4(0, i, shape[0], j, shape[1], k, shape[2], l, shape[3]) - } break; - default: - assert(false && "not implemented"); + if (dataInfo) { + printf("\tdata: ("); + switch (rank) { + case 1: { + int64_t indexes[1]; + LOOP_1(0, i, shape[0]) + } break; + case 2: { + int64_t indexes[2]; + LOOP_2(0, i, shape[0], j, shape[1]) + } break; + case 3: { + int64_t indexes[3]; + LOOP_3(0, i, shape[0], j, shape[1], k, shape[2]) + } break; + case 4: { + int64_t indexes[4]; + LOOP_4(0, i, shape[0], j, shape[1], k, shape[2], l, shape[3]) + } break; + default: + assert(false && "not implemented"); + } + printf(")\n"); } - printf(")\n"); } #ifdef __cplusplus diff --git a/src/Transform/ONNX/CMakeLists.txt b/src/Transform/ONNX/CMakeLists.txt index e7cdcc76ead..4958612716d 100644 --- a/src/Transform/ONNX/CMakeLists.txt +++ b/src/Transform/ONNX/CMakeLists.txt @@ -44,6 +44,7 @@ add_onnx_mlir_library(OMShapeInference add_onnx_mlir_library(OMInstrumentONNX InstrumentONNXPass.cpp + InstrumentONNXSignaturePass.cpp INCLUDE_DIRS PUBLIC ${ONNX_MLIR_SRC_ROOT}/include diff --git a/src/Transform/ONNX/InstrumentONNXSignaturePass.cpp b/src/Transform/ONNX/InstrumentONNXSignaturePass.cpp new file mode 100644 index 00000000000..82c4f21a7e3 --- /dev/null +++ b/src/Transform/ONNX/InstrumentONNXSignaturePass.cpp @@ -0,0 +1,99 @@ +/* + * SPDX-License-Identifier: Apache-2.0 + */ + +//===------- InstrumentONNXSignaturePass.cpp - Instrumentation +//---------------------===// +// +// Copyright 2022 The IBM Research Authors. +// +// ============================================================================= +// +// This file implements a Function level pass that inserts krnl print statements +// that print the operation name and its input type signature at runtime. +// +//===----------------------------------------------------------------------===// + +#include + +#include "onnx-mlir/Compiler/OMCompilerTypes.h" + +#include "mlir/Dialect/Func/IR/FuncOps.h" +#include "mlir/IR/BuiltinTypes.h" +#include "mlir/Interfaces/CallInterfaces.h" +#include "mlir/Pass/Pass.h" +#include "llvm/ADT/SmallPtrSet.h" +#include "llvm/Support/raw_ostream.h" + +#include "src/Dialect/Krnl/DialectBuilder.hpp" +#include "src/Dialect/Krnl/KrnlOps.hpp" +#include "src/Dialect/ONNX/ONNXOps.hpp" +#include "src/Interface/ShapeInferenceOpInterface.hpp" +#include "src/Pass/Passes.hpp" + +using namespace mlir; + +namespace { + +/*! + * This pass insert KrnlPrint and KrnlPrintTensor before each ONNX ops to print + * an operation name and input operand type signatures at runtime. + */ + +class InstrumentONNXSignaturePass + : public mlir::PassWrapper> { + +public: + MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(InstrumentONNXSignaturePass) + + InstrumentONNXSignaturePass() = default; + InstrumentONNXSignaturePass(const InstrumentONNXSignaturePass &pass) + : mlir::PassWrapper>() {} + +private: +public: + StringRef getArgument() const override { + return "instrument-onnx-runtime-signature"; + } + + StringRef getDescription() const override { + return "instrument on onnx ops to print their input operand's type " + "signature"; + } + + void runOnOperation() override { + // Iterate on the operations nested in this function. + getOperation().walk([&](mlir::Operation *op) { + if (isa(op->getDialect())) { + if (!isa(op)) { + Location loc = op->getLoc(); + OpBuilder builder(op); + ValueRange operands = op->getOperands(); + std::string opName( + op->getName().getStringRef().data() + 5); // Skip "onnx.". + StringAttr opNameAttr = builder.getStringAttr(opName); + if (isa(op)) { + // Constant has the type in the output, so use it. + operands = op->getResults(); + // Since we use the result of the constant operation, we must insert + // the print operation after the constant operation. + builder.setInsertionPointAfter(op); + builder.create(loc, opNameAttr, operands); + } else if (operands.size() > 0) { + builder.create(loc, opNameAttr, operands); + } + } + } + }); + } +}; +} // end anonymous namespace + +/*! + * Create an instrumentation pass. + */ +std::unique_ptr onnx_mlir::createInstrumentONNXSignaturePass() { + return std::make_unique(); +} From 21e0ed405252d90a184f01781ec7787a589b4fa8 Mon Sep 17 00:00:00 2001 From: Alexandre Eichenberger Date: Wed, 3 Aug 2022 14:18:35 -0400 Subject: [PATCH 2/5] Vsprint doc v2 (#1574) Signed-off-by: Alexandre Eichenberger --- src/Runtime/jni/jnilog.c | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/Runtime/jni/jnilog.c b/src/Runtime/jni/jnilog.c index f2f9b381b10..c30b9f549c3 100644 --- a/src/Runtime/jni/jnilog.c +++ b/src/Runtime/jni/jnilog.c @@ -169,6 +169,7 @@ void log_printf( * of 8 bytes on z/OS. */ pthread_t tid = get_threadid(); + assert(LOG_MAX_LEN >= strlen(buf) && "error in snprintf length"); snprintf(buf + strlen(buf), LOG_MAX_LEN - strlen(buf), "[%016lx][%s]%s:%s:%d ", *(unsigned long *)&tid, log_level_name[level], get_filename(file), func, line); @@ -196,7 +197,7 @@ void log_printf( As an added security, we added an assert to make sure that quantity is positive. */ - assert(LOG_MAX_LEN - strlen(buf) >= 0 && "error in vsnprintf length"); + assert(LOG_MAX_LEN >= strlen(buf) && "error in vsnprintf length"); va_list log_data; va_start(log_data, fmt); From 022e576e7ecd6853d51acd04d114dbf672e39a4d Mon Sep 17 00:00:00 2001 From: Haruki Imai Date: Thu, 4 Aug 2022 23:38:58 +0900 Subject: [PATCH 3/5] Fix condition where reinterpret_cast is used in lowering of transpose op. (#1576) Changed to use input MemRefType in the condition check to determine if reinterpret_cast can be used in lowering of Transpose op. * Change memRefType for checking condition for reinterpret_cast. Signed-off-by: Haruki Imai * Add a lit test Signed-off-by: Haruki Imai --- .../ONNXToKrnl/Tensor/Transpose.cpp | 28 ++++++++++++------- .../onnx/onnx_lowering_with_canonicalize.mlir | 13 +++++++++ 2 files changed, 31 insertions(+), 10 deletions(-) diff --git a/src/Conversion/ONNXToKrnl/Tensor/Transpose.cpp b/src/Conversion/ONNXToKrnl/Tensor/Transpose.cpp index 24cd649389a..5df287aabb1 100644 --- a/src/Conversion/ONNXToKrnl/Tensor/Transpose.cpp +++ b/src/Conversion/ONNXToKrnl/Tensor/Transpose.cpp @@ -34,12 +34,19 @@ struct ONNXTransposeOpLowering : public ConversionPattern { Value data = operandAdaptor.data(); auto permAttr = transposeOp.perm(); + // Convert the input type to MemRefType. + Type inConvertedType = typeConverter->convertType(data.getType()); + assert(inConvertedType && inConvertedType.isa() && + "Failed to convert type to MemRefType"); + MemRefType inMemRefType = inConvertedType.cast(); + uint64_t inRank = inMemRefType.getShape().size(); // Convert the output type to MemRefType. - Type convertedType = typeConverter->convertType(*op->result_type_begin()); - assert(convertedType && convertedType.isa() && + Type outConvertedType = + typeConverter->convertType(*op->result_type_begin()); + assert(outConvertedType && outConvertedType.isa() && "Failed to convert type to MemRefType"); - MemRefType memRefType = convertedType.cast(); - uint64_t rank = memRefType.getShape().size(); + MemRefType outMemRefType = outConvertedType.cast(); + uint64_t outRank = outMemRefType.getShape().size(); // Get a shape helper. ONNXTransposeOpShapeHelper shapeHelper(&transposeOp, &rewriter, @@ -51,17 +58,18 @@ struct ONNXTransposeOpLowering : public ConversionPattern { // If the order of the dimensions whose value is not 1 does not change after // transpose, it is safe to lower transpose to a view op. - ArrayRef dims = memRefType.getShape(); + ArrayRef dims = inMemRefType.getShape(); SmallVector originalAxes; for (uint64_t axis = 0; axis < dims.size(); ++axis) if (dims[axis] != 1) originalAxes.emplace_back(axis); SmallVector permutedAxes; - for (uint64_t i = 0; i < rank; ++i) { + for (uint64_t i = 0; i < inRank; ++i) { int64_t axis = ArrayAttrIntVal(permAttr, i); if (dims[axis] != 1) permutedAxes.emplace_back(axis); } + if (originalAxes == permutedAxes) { // It is safe to lower to a view op. MemRefBuilder createMemRef(rewriter, loc); @@ -73,11 +81,11 @@ struct ONNXTransposeOpLowering : public ConversionPattern { // Insert an allocation and deallocation for the result of this operation. Value alloc = insertAllocAndDeallocSimple( - rewriter, op, memRefType, loc, shapeHelper.dimsForOutput()); + rewriter, op, outMemRefType, loc, shapeHelper.dimsForOutput()); KrnlBuilder createKrnl(rewriter, loc); - ValueRange loopDef = createKrnl.defineLoops(rank); - SmallVector lbs(rank, LiteralIndexExpr(0)); + ValueRange loopDef = createKrnl.defineLoops(outRank); + SmallVector lbs(outRank, LiteralIndexExpr(0)); MemRefBoundsIndexCapture dataBounds(data); SmallVector ubs; @@ -87,7 +95,7 @@ struct ONNXTransposeOpLowering : public ConversionPattern { [&](KrnlBuilder &createKrnl, ValueRange indices) { // Compute the indices used by the load operation. SmallVector storeIndices; - for (uint64_t i = 0; i < rank; ++i) { + for (uint64_t i = 0; i < outRank; ++i) { Value index = indices[ArrayAttrIntVal(permAttr, i)]; storeIndices.emplace_back(DimIndexExpr(index)); } diff --git a/test/mlir/onnx/onnx_lowering_with_canonicalize.mlir b/test/mlir/onnx/onnx_lowering_with_canonicalize.mlir index d5944a38a28..f4b21eed65d 100644 --- a/test/mlir/onnx/onnx_lowering_with_canonicalize.mlir +++ b/test/mlir/onnx/onnx_lowering_with_canonicalize.mlir @@ -3566,3 +3566,16 @@ func.func @test_transpose_lowered_to_a_view_op(%arg0: tensor) -> // CHECK: } } +// ----- + +// Check lowering transpose to a view op when the order of the dimensions whose +// value is not 1 is unchanged. +// The order of the dimension whose value is not 1 is changed by transpose. +func.func @test_transpose_lowered_to_a_view_op_inv(%arg0: tensor) -> tensor<*xf32> { + %0 = "onnx.Transpose"(%arg0) {perm = [3, 0, 1, 2]} : (tensor) -> tensor<*xf32> + return %0 : tensor<*xf32> + // CHECK-LABEL: func @test_transpose_lowered_to_a_view_op_inv + // CHECK-SAME: ([[PARAM_0_:%.+]]: memref) -> memref<384x?x1x1xf32> { + // CHECK-NOT: memref.reinterpret_cast + +} From a6965a64e9cda9db229fa18a173dcdb07ba00733 Mon Sep 17 00:00:00 2001 From: Soren Lassen Date: Thu, 4 Aug 2022 18:04:07 -0700 Subject: [PATCH 4/5] MatMul 1-D x 1-D results in scalar (#1581) Signed-off-by: Soren Lassen --- src/Dialect/ONNX/ShapeInference/MatMul.cpp | 2 +- .../onnx/onnx_lowering_with_canonicalize.mlir | 20 +++++++++---------- test/mlir/onnx/onnx_shape_inference.mlir | 6 +++--- 3 files changed, 14 insertions(+), 14 deletions(-) diff --git a/src/Dialect/ONNX/ShapeInference/MatMul.cpp b/src/Dialect/ONNX/ShapeInference/MatMul.cpp index 35e1896eff7..0c9486b3923 100644 --- a/src/Dialect/ONNX/ShapeInference/MatMul.cpp +++ b/src/Dialect/ONNX/ShapeInference/MatMul.cpp @@ -134,7 +134,7 @@ LogicalResult ONNXMatMulOpShapeHelper::computeShape( outputDims.emplace_back(bDims[bM]); // For the case where both aRank == bRank == 1 if (ABounds.getRank() == 1 && BBounds.getRank() == 1) { - outputDims.emplace_back(oneIE); + assert(outputDims.empty() && "1-D x 1-D results in scalar"); } // Save the final result. dimsForOutput() = outputDims; diff --git a/test/mlir/onnx/onnx_lowering_with_canonicalize.mlir b/test/mlir/onnx/onnx_lowering_with_canonicalize.mlir index f4b21eed65d..5cc8fda7c85 100644 --- a/test/mlir/onnx/onnx_lowering_with_canonicalize.mlir +++ b/test/mlir/onnx/onnx_lowering_with_canonicalize.mlir @@ -1217,22 +1217,22 @@ func.func private @test_matmul6(%arg0 : tensor, %arg1 : tensor<5xf32 // ----- -// 1-D x 1-D +// 1-D x 1-D results in scalar func.func private @test_matmul7(%arg0 : tensor<5xf32>, %arg1 : tensor<5xf32>) -> tensor<*xf32> { %0 ="onnx.MatMul"(%arg0, %arg1) : (tensor<5xf32>, tensor<5xf32>) -> tensor<*xf32> "func.return"(%0) : (tensor<*xf32>) -> () // mlir2FileCheck.py -a'["A", "B"]' -n'{"1": "RES"}' // CHECK-LABEL: func private @test_matmul7 -// CHECK-SAME: ([[A_:%.+]]: memref<5xf32>, [[B_:%.+]]: memref<5xf32>) -> memref<1xf32> { +// CHECK-SAME: ([[A_:%.+]]: memref<5xf32>, [[B_:%.+]]: memref<5xf32>) -> memref { // CHECK-DAG: [[VAR_cst_:%.+]] = arith.constant 0.000000e+00 : f32 -// CHECK-DAG: [[RES_:%.+]] = memref.alloc() {{.*}}: memref<1xf32> -// CHECK-DAG: [[RES_1_:%.+]]:2 = krnl.define_loops 2 +// CHECK-DAG: [[RES_:%.+]] = memref.alloc() : memref +// CHECK-DAG: [[RES_1_:%.+]] = krnl.define_loops 1 // CHECK-DAG: [[RES_2_:%.+]] = memref.alloca() : memref -// CHECK: krnl.iterate([[RES_1_]]#0) with ([[RES_1_]]#0 -> [[I_0_:%.+]] = 0 to 1, [[RES_1_]]#1 -> [[I_1_:%.+]] = 0 to 5){ -// CHECK: [[VAR_3_:%.+]] = krnl.get_induction_var_value([[RES_1_]]#0) : (!krnl.loop) -> index +// CHECK: krnl.iterate() with ([[RES_1_]] -> [[I_1_:%.+]] = 0 to 5){ +// CHECK: krnl.get_induction_var_value() : () -> () // CHECK: krnl.store [[VAR_cst_]], [[RES_2_]][] : memref -// CHECK: krnl.iterate([[RES_1_]]#1) with (){ -// CHECK: [[VAR_5_:%.+]] = krnl.get_induction_var_value([[RES_1_]]#1) : (!krnl.loop) -> index +// CHECK: krnl.iterate([[RES_1_]]) with (){ +// CHECK: [[VAR_5_:%.+]] = krnl.get_induction_var_value([[RES_1_]]) : (!krnl.loop) -> index // CHECK-DAG: [[LOAD_A_MEM_:%.+]] = krnl.load [[A_]]{{.}}[[VAR_5_]]{{.}} : memref<5xf32> // CHECK-DAG: [[LOAD_B_MEM_:%.+]] = krnl.load [[B_]]{{.}}[[VAR_5_]]{{.}} : memref<5xf32> // CHECK-DAG: [[LOAD_RES_2_MEM_:%.+]] = krnl.load [[RES_2_]][] : memref @@ -1241,9 +1241,9 @@ func.func private @test_matmul7(%arg0 : tensor<5xf32>, %arg1 : tensor<5xf32>) -> // CHECK: krnl.store [[VAR_10_]], [[RES_2_]][] : memref // CHECK: } // CHECK: [[LOAD_RES_2_MEM_1_:%.+]] = krnl.load [[RES_2_]][] : memref -// CHECK: krnl.store [[LOAD_RES_2_MEM_1_]], [[RES_]]{{.}}[[VAR_3_]]{{.}} : memref<1xf32> +// CHECK: krnl.store [[LOAD_RES_2_MEM_1_]], [[RES_]][] : memref // CHECK: } -// CHECK: return [[RES_]] : memref<1xf32> +// CHECK: return [[RES_]] : memref // CHECK: } } diff --git a/test/mlir/onnx/onnx_shape_inference.mlir b/test/mlir/onnx/onnx_shape_inference.mlir index 8d4075a9699..ebfd6919914 100644 --- a/test/mlir/onnx/onnx_shape_inference.mlir +++ b/test/mlir/onnx/onnx_shape_inference.mlir @@ -68,15 +68,15 @@ func.func @test_transpose(%arg0 : tensor<5x5x1x32xf32>) -> tensor<*xf32> { /// Test the shape inferencing scheme for the matmul operation. //===----------------------------------------------------------------------===// -/// MatMul: 1-D x 1-D +/// MatMul: 1-D x 1-D results in scalar func.func @test_matmul_1(%arg0 : tensor<32xf32>, %arg1 : tensor<32xf32>) -> tensor<*xf32> { %0 = "onnx.MatMul"(%arg0, %arg1) : (tensor<32xf32>, tensor<32xf32>) -> tensor<*xf32> "func.return"(%0) : (tensor<*xf32>) -> () // CHECK-LABEL: test_matmul_1 - // CHECK: [[RES1:%.+]] = "onnx.MatMul"(%arg0, %arg1) : (tensor<32xf32>, tensor<32xf32>) -> tensor<1xf32> - // CHECK: return [[RES1]] : tensor<1xf32> + // CHECK: [[RES1:%.+]] = "onnx.MatMul"(%arg0, %arg1) : (tensor<32xf32>, tensor<32xf32>) -> tensor + // CHECK: return [[RES1]] : tensor } // ----- From 8f075bb2d5847a82e82b07d99e906f6f70223ff0 Mon Sep 17 00:00:00 2001 From: Haruki Imai Date: Fri, 5 Aug 2022 21:58:09 +0900 Subject: [PATCH 5/5] [NNPA] Enabled Conv2D C != 1, kH=1, kW=1 on NNPA (#1583) * Revert "[NNPA] Run Conv2D on CPU when input parameters are C != 1, kH = 1, kW=1 (#1518)" Signed-off-by: Haruki Imai --- .../Conversion/ONNXToZHigh/ONNXLegalityCheck.cpp | 7 ------- .../nnpa/conversion/onnx-to-zhigh/conv.mlir | 16 ++-------------- 2 files changed, 2 insertions(+), 21 deletions(-) diff --git a/src/Accelerators/NNPA/Conversion/ONNXToZHigh/ONNXLegalityCheck.cpp b/src/Accelerators/NNPA/Conversion/ONNXToZHigh/ONNXLegalityCheck.cpp index 5338b3d6eec..4a52effe2e8 100644 --- a/src/Accelerators/NNPA/Conversion/ONNXToZHigh/ONNXLegalityCheck.cpp +++ b/src/Accelerators/NNPA/Conversion/ONNXToZHigh/ONNXLegalityCheck.cpp @@ -829,7 +829,6 @@ bool isSuitableForZDNN(ONNXConvOp op) { [](IndexExpr val) { return !val.isLiteral(); })) return false; - int64_t inputShapeC = shapeInput[1]; int64_t inputShapeH = shapeInput[2]; int64_t inputShapeW = shapeInput[3]; int64_t outputShapeH = shapeOutput[2]; @@ -849,12 +848,6 @@ bool isSuitableForZDNN(ONNXConvOp op) { if (!isWOK) return false; - // Currently disable the generation of Conv2D when parameters are C != 1, kH = - // 1, kW=1 because of current issue #1517. When fixed, please remove lit test - // test_onnx_conv2d_not_lowered_c_not_1_kernel11. - if (inputShapeC != 1 && kernelShapeH == 1 && kernelShapeW == 1) - return false; - return true; } diff --git a/test/mlir/accelerators/nnpa/conversion/onnx-to-zhigh/conv.mlir b/test/mlir/accelerators/nnpa/conversion/onnx-to-zhigh/conv.mlir index 58591978302..ce478d0a761 100644 --- a/test/mlir/accelerators/nnpa/conversion/onnx-to-zhigh/conv.mlir +++ b/test/mlir/accelerators/nnpa/conversion/onnx-to-zhigh/conv.mlir @@ -91,8 +91,8 @@ func.func @test_onnx_conv2d_stride_13(%arg0: tensor<5x3x1024x1024xf32>, %arg1 : // ----- -func.func @test_onnx_conv2d_valid_padding_H_equal_KW(%arg0: tensor, %arg1: tensor<448x1280x2x2xf32>, %arg2: tensor<448xf32>) -> tensor<*xf32> { - %0 = "onnx.Conv"(%arg0, %arg1, %arg2) {dilations = [1, 1], group = 1 : si64, kernel_shape = [2, 2], pads = [0, 0, 0, 0], strides = [1, 1]} : (tensor, tensor<448x1280x2x2xf32>, tensor<448xf32>) -> tensor<*xf32> +func.func @test_onnx_conv2d_valid_padding_H_equal_KW(%arg0: tensor, %arg1: tensor<448x1280x1x1xf32>, %arg2: tensor<448xf32>) -> tensor<*xf32> { + %0 = "onnx.Conv"(%arg0, %arg1, %arg2) {dilations = [1, 1], group = 1 : si64, kernel_shape = [1, 1], pads = [0, 0, 0, 0], strides = [1, 1]} : (tensor, tensor<448x1280x1x1xf32>, tensor<448xf32>) -> tensor<*xf32> return %0 : tensor<*xf32> // CHECK-LABEL: test_onnx_conv2d_valid_padding_H_equal_KW // CHECK: "zhigh.Conv2D" @@ -225,15 +225,3 @@ func.func @test_exceed_limit_conv2d(%arg0: tensor<32769x3x32x32xf32>, %arg1 : te // CHECK-LABEL: func @test_exceed_limit_conv2d // CHECK: "onnx.Conv" } - -// ----- - -/// COM: Workaround for issue #1517 -/// COM: Not lowered to zhigh when C != 1, kH = 1, and kW = 1 -func.func @test_onnx_conv2d_not_lowered_c_not_1_kernel11(%arg0: tensor<5x3x32x32xf32>, %arg1 : tensor<2x3x1x1xf32>, %arg2: tensor<2xf32>) -> tensor<*xf32> { - %0 = "onnx.Conv"(%arg0, %arg1, %arg2) {kernel_shape = [1, 1]} : (tensor<5x3x32x32xf32>, tensor<2x3x1x1xf32>, tensor<2xf32>) -> tensor<*xf32> - return %0 : tensor<*xf32> - - // CHECK-LABEL: test_onnx_conv2d_not_lowered_c_not_1_kernel11 - // CHECK: "onnx.Conv" -}