Lines Matching refs:scf
1 // RUN: mlir-opt %s -scf-bufferize | FileCheck %s
9 // CHECK: %[[RESULT_MEMREF:.*]] = scf.if %[[PRED]] -> (memref<?xf32>) {
10 // CHECK: scf.yield %[[TRUE_MEMREF]] : memref<?xf32>
12 // CHECK: scf.yield %[[FALSE_MEMREF]] : memref<?xf32>
18 %0 = scf.if %pred -> (tensor<?xf32>) {
19 scf.yield %true_val : tensor<?xf32>
21 scf.yield %false_val : tensor<?xf32>
31 // CHECK: %[[RESULT_MEMREF:.*]] = scf.for %[[VAL_6:.*]] = %[[LB]] to %[[UB]] step %[[STEP…
32 // CHECK: scf.yield %[[ITER]] : memref<f32>
38 %ret = scf.for %iv = %lb to %ub step %step iter_args(%iter = %arg0) -> tensor<f32> {
39 scf.yield %iter : tensor<f32>
50 %0 = scf.if %pred -> (tensor<f32>) {
52 scf.yield %1: tensor<f32>
55 scf.yield %1 : tensor<f32>
64 // CHECK: %[[RESULT:.*]] = scf.for %[[IV:.*]] = %[[INDEX]] to %[[INDEX]] step %[[INDEX]] …
68 // CHECK: scf.yield %[[MEMREF_MUNGED]] : memref<f32>
74 %ret = scf.for %iv = %index to %index step %index iter_args(%iter = %arg0) -> tensor<f32> {
76 scf.yield %0 : tensor<f32>
84 // CHECK: %[[RES1:.*]]:3 = scf.while (%{{.*}} = %[[ARG0]], %{{.*}} = %[[M]]) : (i64, memref<f32>) -…
85 // CHECK: scf.condition(%{{.*}}) %{{.*}}, %{{.*}}, %{{.*}} : i64, i64, memref<f32>
87 // CHECK: scf.yield %{{.*}}, %{{.*}} : i64, memref<f32>
92 %0:3 = scf.while (%arg3 = %arg0, %arg4 = %arg2) : (i64, tensor<f32>) -> (i64, i64, tensor<f32>) {
94 scf.condition(%1) %arg3, %arg3, %arg4 : i64, i64, tensor<f32>
98 scf.yield %1, %arg7 : i64, tensor<f32>