Home
last modified time | relevance | path

Searched refs:while_cond (Results 1 – 25 of 28) sorted by relevance

12

/external/llvm-project/llvm/test/Transforms/CodeGenPrepare/ARM/
Dlarge-offset-gep.ll9 br label %while_cond
11 while_cond:
24 br label %while_cond
37 br i1 %cmp, label %while_end, label %while_cond
39 while_cond:
52 br label %while_cond
72 br i1 %cmp, label %while_end, label %while_cond
74 while_cond:
87 br label %while_cond
101 %struct = invoke %struct_type* @foo() to label %while_cond unwind label %cleanup
[all …]
Dsplitgep.ll17 br label %while_cond
19 while_cond:
35 br label %while_cond
/external/llvm-project/llvm/test/Transforms/CodeGenPrepare/AArch64/
Dlarge-offset-gep.ll9 br label %while_cond
11 while_cond:
25 br label %while_cond
35 br i1 %cmp, label %while_end, label %while_cond
37 while_cond:
51 br label %while_cond
68 br i1 %cmp, label %while_end, label %while_cond
70 while_cond:
84 br label %while_cond
96 br label %while_cond
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_simplifier.cc53 HloComputation* while_cond = while_op->while_condition(); in RemoveDeadTupleIndices() local
129 while_cond->CloneWithReplacements( in RemoveDeadTupleIndices()
130 make_while_computation_replacements(while_cond)); in RemoveDeadTupleIndices()
225 HloComputation* while_cond = while_op->while_condition(); in TryRemoveDeadWhileParams() local
244 while_cond->parameter_instruction(0)}) { in TryRemoveDeadWhileParams()
265 for (HloComputation* comp : {while_body, while_cond}) { in TryRemoveDeadWhileParams()
337 HloComputation* while_cond = while_op->while_condition(); in TryRemoveRepeatedWhileTupleIndicesHelper() local
348 for (HloComputation* comp : {while_body, while_cond}) { in TryRemoveRepeatedWhileTupleIndicesHelper()
399 HloComputation* while_cond = while_op->while_condition(); in TryRemoveRepeatedWhileTupleIndices() local
479 while_cond = while_op->while_condition(); in TryRemoveRepeatedWhileTupleIndices()
[all …]
Dwhile_loop_analysis.cc252 auto* while_cond = while_op->while_condition(); in GetLoopInductionVarTupleIdx() local
253 auto* while_cond_root = while_cond->root_instruction(); in GetLoopInductionVarTupleIdx()
254 auto* while_cond_param = while_cond->parameter_instruction(0); in GetLoopInductionVarTupleIdx()
259 << while_cond->root_instruction()->ToString(); in GetLoopInductionVarTupleIdx()
406 auto* while_cond = while_op->while_condition(); in PatternMatchLoopTripCount() local
407 auto* while_cond_root = while_cond->root_instruction(); in PatternMatchLoopTripCount()
511 auto* while_cond = while_op->while_condition(); in ComputeWhileLoopTripCount() local
512 auto* while_cond_root = while_cond->root_instruction(); in ComputeWhileLoopTripCount()
580 auto* while_cond = while_op->while_condition(); in ComputeWhileLoopTripCountUpperBound() local
581 auto* while_cond_param = while_cond->parameter_instruction(0); in ComputeWhileLoopTripCountUpperBound()
[all …]
Dwhile_loop_constant_sinking.cc51 HloComputation* while_cond = while_instr->while_condition(); in TrySinkingConstantsIntoWhileLoop() local
63 WhileUtil::GetGTEsMapForWhileConditional(*while_cond); in TrySinkingConstantsIntoWhileLoop()
96 HloInstruction* constant_instr = while_cond->AddInstruction( in TrySinkingConstantsIntoWhileLoop()
Dhlo_rematerialization_test_utils.h102 HloComputation* while_cond, HloComputation* while_body,
116 vec1_shape_, while_cond, while_body, slice_1));
Dhlo_rematerialization_test.cc155 HloComputation* while_cond = in TEST_F() local
162 while_cond, /*while_body=*/body_computation)); in TEST_F()
191 HloComputation* while_cond = in TEST_F() local
198 while_cond, /*while_body=*/body_computation)); in TEST_F()
223 HloComputation* while_cond = in TEST_F() local
226 module->AddEmbeddedComputation(while_cond->Clone()); in TEST_F()
232 while_cond, /*while_body=*/inner_computation, in TEST_F()
Dwhile_loop_simplifier_test.cc234 auto* while_cond = while_op->while_condition(); in TEST_F() local
235 auto token = while_cond->AddInstruction(HloInstruction::CreateToken()); in TEST_F()
236 while_cond->AddInstruction(HloInstruction::CreateInfeed( in TEST_F()
Dhlo_verifier.cc1816 auto* while_cond = xla_while->while_condition(); in HandleWhile() local
1818 if (while_cond->num_parameters() != 1) { in HandleWhile()
1821 while_cond->num_parameters(), while_cond->ToString()); in HandleWhile()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dregion-control-flow-to-functional.mlir542 // CHECK: func private @while_cond
545 func private @while_cond(%arg0 : tensor<*xf32>, %arg1 : tensor<i32>) -> tensor<i1>
548 // CHECK: [[Result:%.*]]:2 = "tf.While"(%arg0, %arg1) {body = @while_body, cond = @while_cond
552 %cond = call @while_cond(%carg0, %carg1) : (tensor<*xf32>, tensor<i32>) -> tensor<i1>
569 // CHECK: func private @while_cond
572 func private @while_cond(%arg0 : tensor<4xf32>, %arg1 : tensor<i32>) -> tensor<i1>
575 // CHECK: [[Result:%.*]]:2 = "tf.While"(%arg0, %arg1) {body = @while_body, cond = @while_cond
580 %cond = call @while_cond(%cond_cast, %carg1) : (tensor<4xf32>, tensor<i32>) -> tensor<i1>
598 // CHECK: func private @while_cond
601 func private @while_cond(%arg0 : tensor<4xf32>, %arg1 : tensor<i32>) -> tensor<i1>
[all …]
Ddrop_while_shape_invariant.mlir6 …%0 = "tf.While"(%arg0) {cond = @while_cond, body = @while_body, is_stateless = false, shape_invari…
21 func @while_cond(%arg0: tensor<*xf32>) -> tensor<i1> {
Dcluster_tf_ops_pass.mlir10 // CHECK-SAME: cond = @while_cond
15 …%1 = "tf.While"(%arg0) {cond = @while_cond, body = @while_body, is_stateless = false, shape_invari…
27 // CHECK: func @while_cond(%[[ARG_0:.*]]: tensor<i32> {tf.device = "/job:localhost/replica:0/task:0…
32 func @while_cond(%arg0: tensor<i32> {tf.device = "/job:localhost/replica:0/task:0/device:CPU:0"}) -…
Dresource_op_lifting.mlir162 {body = @while_body, cond = @while_cond, device = "", is_stateless = false}
192 // CHECK: func @while_cond(%[[CARG0:.*]]: tensor<i32>, %[[CARG1:.*]]: tensor<f32>)
193 func @while_cond(%arg0: tensor<i32>, %arg1: tensor<*x!tf.resource<tensor<f32>>>, %arg2: tensor<*x!t…
211 body = @while_body, cond = @while_cond, device = "", is_stateless = false}
230 // CHECK: func @while_cond(%arg0: tensor<f32>)
231 func @while_cond(%arg0: tensor<*x!tf.resource<tensor<f32>>>) -> tensor<f32> {
252 body = @while_body, cond = @while_cond, device = "", is_stateless = false}
268 // CHECK: func @while_cond(%[[CARG0:.*]]: tensor<f32>)
269 func @while_cond(%arg0: tensor<*x!tf.resource<tensor<f32>>>) -> tensor<f32> {
290 body = @while_body, cond = @while_cond, device = "", is_stateless = false}
[all …]
Dresource-alias-analysis-test.mlir170 body = @while_body, cond = @while_cond, is_stateless = false
188 // CHECK-LABEL: func @while_cond
192 func @while_cond(%arg0: !tf_res, %arg1: !tf_res, %arg2: !tf_res) -> tensor<i1> {
Dresource-device-inference.mlir134 cond = @while_cond, is_stateless = false}
177 // CHECK-LABEL: func @while_cond
178 func @while_cond(
Dside-effect-analysis-test.mlir461 body = @while_body, cond = @while_cond, is_stateless = true}
502 // CHECK-LABEL: func @while_cond
503 func @while_cond(%arg0: tensor<i1>) -> tensor<i1> {
930 body = @while_body, cond = @while_cond, is_stateless = false}
1020 // CHECK-LABEL: func @while_cond
1021 func @while_cond(
Dstack_ops_decomposition.mlir94 body = @while_body, cond = @while_cond, device = "", is_stateless = false}
117 // CHECK: func @while_cond(%[[CARG0:.*]]: tensor<!tf.resource<tensor<10xf32>>>, %[[CARG1:.*]]: tens…
118 func @while_cond(%arg0: tensor<!tf.resource>, %arg1: tensor<i32>) -> tensor<i32> {
Dtensor_list_ops_decomposition.mlir172 body = @while_body, cond = @while_cond, device = "", is_stateless = false}
196 // CHECK: func @while_cond(%[[CARG0:.*]]: tensor<10xf32>, %[[CARG1:.*]]: tensor<i32>, %[[CARG2:.*]]…
197 func @while_cond(%arg0: tensor<!tf.variant<tensor<f32>>>, %arg1: tensor<i32>) -> tensor<i32> {
/external/tensorflow/tensorflow/python/autograph/operators/
Dcontrol_flow_deprecated_py2.py413 def while_cond(iterate_index, *loop_vars): function
423 while_cond,
477 def while_cond(iterate_index, *loop_vars): function
489 while_cond,
532 def while_cond(iterate, *loop_vars): function
549 while_cond,
614 def while_cond(has_next, *loop_vars): function
624 while_cond,
/external/python/pycparser/examples/
Dexplore_ast.py139 while_cond = while_stmt.cond variable
/external/llvm-project/llvm/test/CodeGen/RISCV/
Dsplit-offsets.ll102 br label %while_cond
103 while_cond:
118 br label %while_cond
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlegalize-tf-while.mlir44 // CHECK: while_cond
54 // INLINE-NOT: while_cond
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dgraph-input-func-arg-name-collision.pbtxt8 # CHECK: func private @while_cond
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-control-flow.mlir169 // CHECK: [[VAL10:%.+]] = call @while_cond([[VAL7]], [[VAL8]], [[VAL9]])
184 …%2:3 = "tf.While"(%0, %1, %0) {body = @while_body, cond = @while_cond, is_stateless = true, parall…
187 func @while_cond(%arg0: tensor<i32>, %arg1: tensor<i32>, %arg2: tensor<i32>) -> tensor<i1> {

12