Home
last modified time | relevance | path

Searched refs:iter_args (Results 1 – 25 of 38) sorted by relevance

12

/external/llvm-project/mlir/test/Transforms/
Dscf-loop-unroll.mlir8 %sum = scf.for %iv = %from to %to step %step iter_args(%sum_iter = %arg0) -> (f32) {
12 …// CHECK: %[[SUM:.*]] = scf.for %{{.*}} = %{{.*}} to %{{.*}} step %{{.*}} iter_args(%[[V0:.*]…
28 %sum:2 = scf.for %iv = %from to %n step %step iter_args(%i0 = %arg0, %i1 = %arg1) -> (f32, f32) {
39 // CHECK-NEXT: %[[SUM:.*]]:2 = scf.for {{.*}} = %[[C0]] to %[[UB]] step %[[C3]] iter_args
41 …// CHECK-NEXT: %[[SUM1:.*]]:2 = scf.for {{.*}} = %[[UB]] to %[[N]] step %[[C1]] iter_args(%[[V1:.*…
Dbuffer-loop-hoisting.mlir143 iter_args(%iterBuf = %buf) -> memref<2xf32> {
170 iter_args(%iterBuf = %buf) -> memref<2xf32> {
203 iter_args(%iterBuf = %buf) -> memref<2xf32> {
205 iter_args(%iterBuf2 = %iterBuf) -> memref<2xf32> {
207 iter_args(%iterBuf3 = %iterBuf2) -> memref<2xf32> {
250 iter_args(%iterBuf = %buf) -> memref<?xf32> {
252 iter_args(%iterBuf2 = %iterBuf) -> memref<?xf32> {
254 iter_args(%iterBuf3 = %iterBuf2) -> memref<?xf32> {
292 iter_args(%iterBuf = %buf) -> memref<2xf32> {
314 iter_args(%iterBuf = %buf) -> memref<2xf32> {
[all …]
Dscf-loop-utils.mlir9 …// CHECK: %[[RES:.*]] = scf.for %[[I:.*]] = %[[lb]] to %[[ub]] step %[[step]] iter_args(%[[VAL:.*]…
28 …// CHECK: %[[RES:.*]]:2 = scf.for %[[I:.*]] = %[[lb]] to %[[ub]] step %[[step]] iter_args(%[[VAL0:…
33 %0 = scf.for %i = %lb to %ub step %step iter_args(%iter = %extra_arg) -> (f32) {
Dsccp-structured.mlir84 %result = scf.for %i0 = %arg0 to %arg1 step %arg2 iter_args(%si = %s0) -> (i32) {
100 %result = scf.for %i0 = %arg0 to %arg1 step %arg2 iter_args(%si = %s0) -> (i32) {
119 %result = scf.for %i0 = %arg0 to %arg1 step %arg2 iter_args(%si = %cst_1) -> (i32) {
Dbuffer-hoisting.mlir623 iter_args(%iterBuf = %buf) -> memref<2xf32> {
650 iter_args(%iterBuf = %buf) -> memref<2xf32> {
682 iter_args(%iterBuf = %buf) -> memref<2xf32> {
684 iter_args(%iterBuf2 = %iterBuf) -> memref<2xf32> {
686 iter_args(%iterBuf3 = %iterBuf2) -> memref<2xf32> {
724 iter_args(%iterBuf = %buf) -> memref<?xf32> {
726 iter_args(%iterBuf2 = %iterBuf) -> memref<?xf32> {
728 iter_args(%iterBuf3 = %iterBuf2) -> memref<?xf32> {
Dbuffer-deallocation.mlir880 iter_args(%iterBuf = %buf) -> memref<2xf32> {
893 // CHECK: %[[ALLOC2:.*]] = scf.for {{.*}} iter_args
923 iter_args(%iterBuf = %buf) -> memref<2xf32> {
937 // CHECK-NEXT: %[[ALLOC1:.*]] = scf.for {{.*}} iter_args(%[[IALLOC:.*]] =
963 iter_args(%iterBuf = %buf) -> memref<2xf32> {
979 // CHECK-NEXT: %[[ALLOC2:.*]] = scf.for {{.*}} iter_args
1018 iter_args(%iterBuf = %buf) -> memref<2xf32> {
1020 iter_args(%iterBuf2 = %iterBuf) -> memref<2xf32> {
1022 iter_args(%iterBuf3 = %iterBuf2) -> memref<2xf32> {
1045 // CHECK-NEXT: %[[VAL_7:.*]] = scf.for {{.*}} iter_args
[all …]
Dpromote-buffers-to-stack.mlir484 iter_args(%iterBuf = %buf) -> memref<2xf32> {
514 iter_args(%iterBuf = %buf) -> memref<2xf32> {
528 // CHECK-NEXT: %[[ALLOCA1:.*]] = scf.for {{.*}} iter_args(%[[IALLOCA:.*]] =
549 iter_args(%iterBuf = %buf) -> memref<2xf32> {
/external/llvm-project/mlir/test/Dialect/Linalg/
Dtile-tensors.mlir10 // CHECK: %[[TD0:.*]] = scf.for {{.*}} to {{.*}} step {{.*}} iter_args(%[[TC0:.*]] = %[[TC]]) …
11 // CHECK: %[[TD1:.*]] = scf.for {{.*}} to {{.*}} step {{.*}} iter_args(%[[TC1:.*]] = %[[TC0]…
12 // CHECK: %[[TD2:.*]] = scf.for {{.*}} to {{.*}} step {{.*}} iter_args(%[[TC2:.*]] = %[[TC…
Dtile-and-fuse-tensors.mlir22 %3 = scf.for %arg3 = %c0 to %0 step %c2 iter_args(%arg4 = %arg2) -> (tensor<?x?xf32>) {
23 %4 = scf.for %arg5 = %c0 to %2 step %c3 iter_args(%arg6 = %arg4) -> (tensor<?x?xf32>) {
24 %5 = scf.for %arg7 = %c0 to %1 step %c4 iter_args(%arg8 = %arg6) -> (tensor<?x?xf32>) {
49 // CHECK-NEXT: scf.for %[[K:[0-9a-z]*]] {{.*}} iter_args(%[[RES:[0-9a-z]*]]
Dhoisting.mlir105 // VECTOR_TRANSFERS: scf.for %[[I:.*]] = %[[LB]] to %[[UB]] step %[[STEP]] iter_args({{.*}}) -> (ve…
107 // VECTOR_TRANSFERS: scf.for %[[J:.*]] = %[[LB]] to %[[UB]] step %[[STEP]] iter_args({{.*}}) -> (…
179 // VECTOR_TRANSFERS: scf.for %[[I:.*]] = %[[LB]] to %[[UB]] step %[[STEP]] iter_args({{.*}}) ->
181 // VECTOR_TRANSFERS: scf.for %[[J:.*]] = %[[LB]] to %[[UB]] step %[[STEP]] iter_args({{.*}}) ->
Dtile-and-distribute.mlir192 // CHECK: %[[TD0:.*]] = scf.for {{.*}} to {{.*}} step {{.*}} iter_args(%[[TC0:.*]] = %[[TC]]) …
195 // CHECK: %[[TD1:.*]] = scf.for {{.*}} to {{.*}} step {{.*}} iter_args(%[[TC1:.*]] = %[[TC0]…
196 // CHECK: %[[TD2:.*]] = scf.for {{.*}} to {{.*}} step {{.*}} iter_args(%[[TC2:.*]] = %[[TC…
/external/llvm-project/mlir/test/Dialect/SCF/
Dbufferize.mlir31 …MEMREF:.*]] = scf.for %[[VAL_6:.*]] = %[[LB]] to %[[UB]] step %[[STEP]] iter_args(%[[ITER:.*]] = %…
40 %ret = scf.for %iv = %lb to %ub step %step iter_args(%iter = %arg0) -> tensor<f32> {
66 …LT:.*]] = scf.for %[[IV:.*]] = %[[INDEX]] to %[[INDEX]] step %[[INDEX]] iter_args(%[[MEMREF_ITER:.…
76 %ret = scf.for %iv = %index to %index step %index iter_args(%iter = %arg0) -> tensor<f32> {
Dcanonicalize.mlir117 %b = scf.for %i = %lb to %ub step %step iter_args(%0 = %a) -> i32 {
130 …%r:3 = scf.for %i = %lb to %ub step %step iter_args(%0 = %a, %1 = %a, %2 = %b) -> (i32, i32, i32) {
140 // CHECK-NEXT: %[[r1:.*]] = scf.for {{.*}} iter_args(%arg4 = %[[a]]) -> (i32) {
211 %0 = scf.for %i = %c42 to %c1 step %c1 iter_args(%arg = %init) -> (i32) {
227 %0 = scf.for %i = %arg0 to %arg0 step %c2 iter_args(%arg = %init) -> (i32) {
246 %0 = scf.for %i = %c42 to %c43 step %c1 iter_args(%arg = %init) -> (i32) {
265 %0 = scf.for %i = %c42 to %c47 step %c5 iter_args(%arg = %init) -> (i32) {
Dops.mlir162 %result = scf.for %i0 = %arg0 to %arg1 step %arg2 iter_args(%si = %s0) -> (f32) {
174 // CHECK-SAME: iter_args(%[[ITER:.*]] = %[[INIT]]) -> (f32) {
184 …%result1:3 = scf.for %i0 = %arg0 to %arg1 step %arg2 iter_args(%si = %s0, %ti = %t0, %ui = %u0) ->…
200 // CHECK-SAME: iter_args(%[[ITER1:.*]] = %[[INIT1]], %[[ITER2:.*]] = %[[INIT2]], %[[ITER3:.*]] = %[…
210 %sum = scf.for %iv = %lb to %ub step %step iter_args(%sum_iter = %sum_0) -> (f32) {
231 // CHECK-SAME: iter_args(%[[ITER:.*]] = %[[INIT]]) -> (f32) {
Dinvalid.mlir358 iter_args(%si = %s0, %ti = %t0) -> (f32, i32, f32) {
374 iter_args(%si = %s0, %ti = %t0, %ui = %u0) -> (f32, i32) {
391 iter_args(%si = %s0, %ti = %t0) -> (i32, i32) {
406 iter_args(%si = %s0, %ti = %t0) -> (f32, f32) {
/external/llvm-project/mlir/integration_test/Dialect/Vector/CPU/
Dtest-maskedstore.mlir20 iter_args(%m_iter = %m) -> (vector<16xf32>) {
44 iter_args(%v_iter = %v) -> (vector<16xf32>) {
Dtest-compress.mlir20 iter_args(%m_iter = %m) -> (vector<16xf32>) {
39 iter_args(%v_iter = %v) -> (vector<16xf32>) {
Dtest-scatter.mlir21 iter_args(%m_iter = %m) -> (vector<8xf32>) {
/external/llvm-project/mlir/test/Dialect/Affine/
Dops.mlir197 %res = affine.for %i = 0 to 10 step 2 iter_args(%sum_iter = %sum_init_0) -> f32 {
210 // CHECK-NEXT: %[[output:.*]] = affine.for %{{.*}} = 0 to 10 step 2 iter_args(%{{.*}} = %[[const_0]…
223 …%res1, %res2 = affine.for %i = 0 to 10 step 2 iter_args(%iter_arg1 = %init_0, %iter_arg2 = %init_0…
232 // CHECK-NEXT: %[[output:[0-9]+]]:2 = affine.for %{{.*}} = 0 to 10 step 2 iter_args(%[[iter_arg1:.*…
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_parallel_loops.cc547 SmallVector<Value, 4> iter_args(rank, zero); in InsertWindowLoops() local
548 iter_args.push_back(b->create<mlir::ConstantOp>( in InsertWindowLoops()
550 iter_args.push_back(b->create<mlir::ConstantOp>( in InsertWindowLoops()
560 b->create<scf::ForOp>(loc, zero, upper, one, iter_args); in InsertWindowLoops()
568 iter_args = ValueRange{result.inner_loop.getRegionIterArgs()}; in InsertWindowLoops()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo-legalize-select-and-scatter.mlir66 // CHECK-SAME: iter_args(
75 // CHECK-SAME: iter_args(
152 // We are in the pad area, return current iter_args.
Dlegalize_to_scf.mlir38 // CHECK: scf.for %[[VAL_21:.*]] = %[[VAL_15]] to %[[VAL_17]] step %[[VAL_19]] iter_args(%[[VAL_22…
/external/llvm-project/mlir/test/Conversion/ShapeToStandard/
Dshape-to-standard.mlir182 // CHECK-NEXT: %[[RESULT:.*]] = scf.for %[[I:.*]] = %[[C0]] to %[[RANK]] step %[[C1]] iter_args(%[[…
279 …EQ_INNER:.*]] = scf.for %[[I:.*]] = %[[C0]] to %[[RANK_A]] step %[[C1]] iter_args(%[[CONJ:.*]] = %…
409 …] = scf.for %[[I:.*]] = %[[RANK_DIFF]] to %[[LARGER_RANK]] step %[[C1]] iter_args(%[[ALL_SO_FAR:.*…
447 …cf.for %[[VAL_16:.*]] = %[[RANK_DIFF]] to %[[LARGER_RANK]] step %[[C1]] iter_args(%[[ALL_SO_FAR:.*…
/external/tensorflow/tensorflow/compiler/mlir/tfr/tests/
Dcontrol_flow.mlir16 %reduce = scf.for %i = %step to %end step %step iter_args(%reduce_iter=%temp) -> !tfr.tensor {
/external/llvm-project/mlir/test/Conversion/GPUToSPIRV/
Dloop.mlir85 … %result:2 = scf.for %i0 = %lb to %ub step %step iter_args(%si = %s0, %sj = %s1) -> (f32, f32) {

12