Lines Matching refs:bufferization

10   %clone = bufferization.clone %buf : memref<*xf32> to memref<*xf32>
17 %0 = bufferization.to_memref %arg0
19 %1 = bufferization.to_memref %arg1
26 %tensor = bufferization.to_tensor %buf restrict writable : memref<2xf32> to tensor<2xf32>
34 // CHECK: bufferization.alloc_tensor(%{{.*}}) : tensor<?x5xf32>
35 %0 = bufferization.alloc_tensor(%sz) : tensor<?x5xf32>
36 // CHECK: bufferization.alloc_tensor() copy(%{{.*}}) : tensor<?x5xf32>
37 %1 = bufferization.alloc_tensor() copy(%t) : tensor<?x5xf32>
38 // CHECK: bufferization.alloc_tensor() : tensor<5x6xf32>
39 %2 = bufferization.alloc_tensor() : tensor<5x6xf32>
40 // CHECK: bufferization.alloc_tensor(%{{.*}}, %{{.*}}) : tensor<?x?xf32>
41 %3 = bufferization.alloc_tensor(%sz, %sz) : tensor<?x?xf32>
42 // CHECK: bufferization.alloc_tensor() copy(%{{.*}}) {escape = true} : tensor<?x5xf32>
43 %4 = bufferization.alloc_tensor() copy(%t) {escape = true} : tensor<?x5xf32>
44 // CHECK: bufferization.alloc_tensor() copy(%{{.*}}) {escape = false} : tensor<?x5xf32>
45 %5 = bufferization.alloc_tensor() copy(%t) {escape = false} : tensor<?x5xf32>
47 // CHECK: bufferization.alloc_tensor() size_hint=
48 %6 = bufferization.alloc_tensor() size_hint=%c100 : tensor<100x100xf64, #CSR>
49 // CHECK: bufferization.alloc_tensor(%{{.+}}) {memory_space = "foo"} : tensor<?xf32>
50 %7 = bufferization.alloc_tensor(%sz) {memory_space = "foo"} : tensor<?xf32>
56 // CHECK: bufferization.dealloc_tensor {{.*}} : tensor<4xi32>
57 bufferization.dealloc_tensor %arg0 : tensor<4xi32>
65 // CHECK: bufferization.materialize_in_destination {{.*}} : (tensor<?xf32>, tensor<?xf32>) -> tensor<?xf32>
66 %1 = bufferization.materialize_in_destination %arg0 in %arg1 : (tensor<?xf32>, tensor<?xf32>) -> tensor<?xf32>
67 // CHECK: bufferization.materialize_in_destination {{.*}} : (tensor<?xf32>, memref<?xf32, 3>) -> ()
68 bufferization.materialize_in_destination %arg0 in restrict writable %arg2 : (tensor<?xf32>, memref<?xf32, 3>) -> ()
69 // CHECK: bufferization.materialize_in_destination {{.*}} : (tensor<?xf32>, tensor<5xf32>) -> tensor<5xf32>
70 %2 = bufferization.materialize_in_destination %arg0 in %arg4 : (tensor<?xf32>, tensor<5xf32>) -> tensor<5xf32>
78 // CHECK: bufferization.dealloc (%arg0, %arg1 : memref<2xf32>, memref<4xi32>) if (%arg2, %arg3) retain (%arg4, %arg5 : memref<?xf32>, memref<*xf64>)
79 %0:2 = bufferization.dealloc (%arg0, %arg1 : memref<2xf32>, memref<4xi32>) if (%arg2, %arg3) retain (%arg4, %arg5 : memref<?xf32>, memref<*xf64>)
80 // CHECK: bufferization.dealloc (%arg0 : memref<2xf32>) if (%arg2)
81 bufferization.dealloc (%arg0 : memref<2xf32>) if (%arg2)
82 // CHECK: bufferization.dealloc
83 bufferization.dealloc