Lines Matching full:shape

1 // RUN: mlir-opt -allow-unregistered-dialect -split-input-file -remove-shape-constraints -canonical…
2 // RUN: mlir-opt -allow-unregistered-dialect -split-input-file -remove-shape-constraints <%s | File…
8 func @f(%arg0 : !shape.shape, %arg1 : !shape.shape) -> index {
9   // REPLACE-NEXT: %[[WITNESS:.+]] = shape.const_witness true
10   // REPLACE-NOT: shape.cstr_eq
11   // REPLACE: shape.assuming %[[WITNESS]]
14   %0 = shape.cstr_broadcastable %arg0, %arg1 : !shape.shape, !shape.shape
15   %1 = shape.assuming %0 -> index {
17     shape.assuming_yield %2 : index
26 func @f(%arg0 : !shape.shape, %arg1 : !shape.shape) -> index {
27   // REPLACE-NEXT: %[[WITNESS:.+]] = shape.const_witness true
28   // REPLACE-NOT: shape.cstr_eq
29   // REPLACE: shape.assuming %[[WITNESS]]
32   %0 = shape.cstr_eq %arg0, %arg1
33   %1 = shape.assuming %0 -> index {
35     shape.assuming_yield %2 : index
45 func @f(%arg0 : !shape.shape, %arg1 : !shape.shape) -> index {
48   %0 = shape.cstr_broadcastable %arg0, %arg1 : !shape.shape, !shape.shape
49   %1 = shape.cstr_eq %arg0, %arg1
50   %2 = shape.assuming_all %0, %1
51   %3 = shape.assuming %0 -> index {
53     shape.assuming_yield %4 : index