We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5466610 commit a85fb9dCopy full SHA for a85fb9d
test/mlir/test/gc/Transforms/DecomposeAggregatedOps.mlir
@@ -0,0 +1,11 @@
1
+// RUN: gc-opt %s -decompose-aggregated-ops | FileCheck %s
2
+
3
+// CHECK-LABEL: softmax
4
+func.func @softmax(%arg0: tensor<2x2x2x2xf32>) -> tensor<2x2x2x2xf32> {
5
+ %0 = tensor.empty() : tensor<2x2x2x2xf32>
6
+ // CHECK-NOT: linalg.softmax
7
+ // CHECK-COUNT-4: linalg.generic
8
+ %1 = linalg.softmax dimension(3)
9
+ ins(%arg0 : tensor<2x2x2x2xf32>) outs(%0 : tensor<2x2x2x2xf32>) -> tensor<2x2x2x2xf32>
10
+ return %1 : tensor<2x2x2x2xf32>
11
+}
0 commit comments