Home
last modified time | relevance | path

Searched refs:else_branch (Results 1 – 25 of 79) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dmerge_control_flow.cc85 for (Operation& op : source.else_branch().front()) { in SafeToMerge()
101 for (Operation& op : destination.else_branch().front()) { in SafeToMerge()
165 Block& destination_else_block = destination.else_branch().front(); in MoveBranches()
166 auto& source_else_body = source.else_branch().front().getOperations(); in MoveBranches()
223 new_if_op.else_branch().push_back(new Block); in CreateMergedIf()
233 destination.else_branch().front().getTerminator()->getOperand( in CreateMergedIf()
235 source.else_branch()); in CreateMergedIf()
268 destination.else_branch().front().getTerminator()->getOperand(i)); in CreateMergedIf()
271 source.else_branch().front().getTerminator()->getOperand(i)); in CreateMergedIf()
272 builder.setInsertionPointToEnd(&new_if_op.else_branch().front()); in CreateMergedIf()
[all …]
Dtpu_extract_outside_compilation.cc142 auto& else_branch = host_side_if.else_branch(); in CloneEmptyIfWithPredicate() local
143 else_branch.push_back(new Block); in CloneEmptyIfWithPredicate()
144 builder.setInsertionPointToEnd(&else_branch.front()); in CloneEmptyIfWithPredicate()
368 MoveOpsToHost(tpu_cluster, &if_op.else_branch().front(), in DecomposeControlFlow()
369 host_if.else_branch().front().getTerminator(), in DecomposeControlFlow()
Dregion_control_flow_to_functional.cc293 IsSingleCallRegion(if_region.else_branch()), in ConvertIfOp()
305 CollectExternValues(if_region.then_branch(), if_region.else_branch()); in ConvertIfOp()
332 ExtractSingleBlockRegion(if_region.else_branch(), else_name, extern_values, in ConvertIfOp()
Dcollection_ops_util.cc184 auto else_branch = if_op.else_function(); in GetElementTypeFromAccess() local
185 assert(then_branch && else_branch); in GetElementTypeFromAccess()
191 else_branch.getArgument(use.getOperandNumber() - 1), module, in GetElementTypeFromAccess()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DIf.pbtxt36 name: "else_branch"
74 name: "else_branch"
111 name: "else_branch"
148 name: "else_branch"
186 name: "else_branch"
DStatelessIf.pbtxt34 name: "else_branch"
71 name: "else_branch"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DIf.pbtxt36 name: "else_branch"
74 name: "else_branch"
111 name: "else_branch"
148 name: "else_branch"
186 name: "else_branch"
DStatelessIf.pbtxt34 name: "else_branch"
71 name: "else_branch"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dfunctional-control-flow-to-regions.mlir12 then_branch = @testIf1Then, else_branch = @testIf1Else, is_stateless = false,
26 // CHECK-NOT: else_branch
45 then_branch = @testIf1Then, else_branch = @testIf1Else, is_stateless = false
69 then_branch = @testIf1Then, else_branch = @testIf1Else, is_stateless = false
91 then_branch = @testIf1Then, else_branch = @testIf1Else, is_stateless = false
115 then_branch = @testIf1Then, else_branch = @testIf1Else, is_stateless = false
139 then_branch = @testIf1Then, else_branch = @testIf1Else, is_stateless = false
255 func private @else_branch() -> ()
260 …"tf.If"(%arg0) {then_branch = @then_branch, else_branch = @else_branch, is_stateless = false, devi…
Dregion-control-flow-to-functional.mlir12 // CHECK-SAME: else_branch = @test_else_name
35 // CHECK-SAME: else_branch = @tf.IfRegion_else
59 // CHECK: "tf.If"{{.+}}else_branch = @tf.IfRegion_else{{.+}}then_branch = @tf.IfRegion_then
80 // CHECK: "tf.If"(%arg0) {else_branch = @tf.IfRegion_else{{.+}}then_branch = @tf.IfRegion_then
100 // CHECK-NEXT: "tf.If"({{.+}}) {else_branch = @tf.IfRegion_else, {{.+}} then_branch = @tf.IfRegion_…
108 …// CHECK: "tf.If"({{.+}}) {else_branch = @tf.IfRegion1_else, {{.+}} then_branch = @tf.IfRegion1_th…
139 // CHECK: "tf.If"({{.+}}) {else_branch = @testIf1Else, {{.+}} then_branch = @testIf1Then}
157 // CHECK: "tf.If"({{.+}}) {else_branch = @testIf1Else, {{.+}} then_branch = @testIf1Then}
177 // CHECK: "tf.If"({{.+}}) {else_branch = @testIf1Else, {{.+}} then_branch = @testIf1Then}
200 …// CHECK: "tf.If"(%arg0, %[[CAST]]) {else_branch = @testIf1Else, {{.+}} then_branch = @testIf1Then}
[all …]
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_If.pbtxt29 types are the same as what else_branch returns.
33 name: "else_branch"
39 summary: "output = cond ? then_branch(input) : else_branch(input)"
Dapi_def_StatelessIf.pbtxt32 types are the same as what else_branch returns.
36 name: "else_branch"
42 summary: "output = cond ? then_branch(input) : else_branch(input)"
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond_test.cc89 StateMap::CondId else_branch; in TEST_F() local
93 else_branch = GetUniqueId(ss); in TEST_F()
97 Status status = JoinCondStatesNonMerge(then_branch, else_branch).status(); in TEST_F()
101 auto joined_or = JoinCondStatesMerge(m, then_branch, else_branch); in TEST_F()
/external/tensorflow/tensorflow/python/framework/
Dauto_control_deps_test.py235 def else_branch(): function
239 constant_op.constant(True), then_branch, else_branch)
250 def else_branch(): function
254 constant_op.constant(False), then_branch, else_branch)
355 def else_branch(): function
359 constant_op.constant(True), then_branch, else_branch)
426 def else_branch(): function
430 constant_op.constant(True), then_branch, else_branch)
441 def else_branch(): function
446 constant_op.constant(False), then_branch, else_branch)
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/
Dif_op.mlir4 // CHECK: %{{.*}} = "tf.If"(%{{.*}}, %{{.*}}, %{{.*}}) {else_branch = @cond_false, is_stateless =…
6 …%1 = "tf.If"(%0, %arg0, %arg1) {else_branch = @cond_false, then_branch = @cond_true, is_stateless …
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Doptimize_functional_ops.mlir12 …%3 = "tf.If"(%2, %0, %1) {else_branch = @sub, then_branch = @add, is_stateless = true} : (tensor<i…
39 …%3 = "tf.If"(%2, %0, %1) {else_branch = @sub, then_branch = @addormul, is_stateless = true} : (ten…
45 …%1 = "tf.If"(%0, %arg1, %arg0) {else_branch = @mul, then_branch = @add, is_stateless = true} : (te…
78 else_branch = @_functionalize_if_else_branch_00, is_stateless = false,
111 else_branch = @_functionalize_if_else_branch_01, is_stateless = false,
147 else_branch = @_functionalize_if_else_branch_02, is_stateless = true,
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/end2end/
Dif_op.pbtxt118 key: "else_branch"
166 key: "else_branch"
412 # CHECK: "tf.If"{{.+}}else_branch = @cond_false_10{{.+}}is_stateless = true{{.+}}then_branch =…
413 # CHECK: "tf.If"{{.+}}else_branch = @cond_false0{{.+}}is_stateless = false{{.+}}then_branch = …
/external/mesa3d/src/gallium/drivers/lima/ir/pp/
Dnir.c642 ppir_branch_node *else_branch, *after_branch; in ppir_emit_if() local
652 else_branch = ppir_node_to_branch(node); in ppir_emit_if()
653 ppir_node_add_src(block->comp, node, &else_branch->src[0], in ppir_emit_if()
655 else_branch->num_src = 1; in ppir_emit_if()
667 else_branch->negate = true; in ppir_emit_if()
668 list_addtail(&else_branch->node.list, &block->node_list); in ppir_emit_if()
677 else_branch->target = ppir_get_block(comp, nblock->successors[0]); in ppir_emit_if()
683 else_branch->target = ppir_get_block(comp, nir_if_first_else_block(if_stmt)); in ppir_emit_if()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/
Dfunctional-if-ops.mlir6 …ontrol_5 = tf_executor.island wraps "tf.If"(%outputs_2, %arg2, %arg3) {else_branch = @cond_false, …
7 …ontrol_7 = tf_executor.island wraps "tf.If"(%outputs_2, %arg2, %arg3) {else_branch = @cond_false, …
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dif_conditional.hlotxt12 %else_branch {
45 …%tuple.5, %tuple.5), true_computation=%then_branch, false_computation=%else_branch, metadata={op_t…
/external/llvm/test/Transforms/RewriteStatepointsForGC/
Drelocation.ll78 br i1 %cnd, label %if_branch, label %else_branch
87 else_branch: ; preds = %bci_0
88 ; CHECK-LABEL: else_branch:
95 join: ; preds = %else_branch, %if_branch
99 ; CHECK-DAG: [ %arg.relocated2, %else_branch ]
/external/llvm-project/llvm/test/Transforms/RewriteStatepointsForGC/
Drelocation.ll79 br i1 %cnd, label %if_branch, label %else_branch
88 else_branch: ; preds = %bci_0
89 ; CHECK-LABEL: else_branch:
96 join: ; preds = %else_branch, %if_branch
100 ; CHECK-DAG: [ %arg.relocated2, %else_branch ]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dfunctional-if-ops.pbtxt54 key: "else_branch"
102 key: "else_branch"
/external/mesa3d/src/gallium/drivers/r600/sfn/
Dsfn_liverange.cpp190 scope_type == else_branch || in is_conditional()
197 if (scope_type == else_branch) in in_else_scope()
217 scope_type == else_branch) in in_ifelse_scope()
887 cur_scope = scopes->create(cur_scope->parent(), else_branch, in scope_else()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_control_flow.cc266 Value else_input = TupleImplicitInputs(op.else_branch(), loc, &builder); in LowerIfRegion()
267 ReplaceTerminator(&op.else_branch().front(), /*extra_results=*/{}, &builder); in LowerIfRegion()
276 if_op.false_branch().takeBody(op.else_branch()); in LowerIfRegion()

1234