| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | merge_control_flow.cc | 82 for (Operation& op : source.then_branch().front()) { in SafeToMerge()
96 for (Operation& op : destination.then_branch().front()) { in SafeToMerge()
159 Block& destination_then_block = destination.then_branch().front(); in MoveBranches()
160 auto& source_then_body = source.then_branch().front().getOperations(); in MoveBranches()
222 new_if_op.then_branch().push_back(new Block); in CreateMergedIf()
228 destination.then_branch().front().getTerminator()->getOperand( in CreateMergedIf()
230 source.then_branch()); in CreateMergedIf()
256 destination.then_branch().front().getTerminator()->getOperand(i)); in CreateMergedIf()
259 source.then_branch().front().getTerminator()->getOperand(i)); in CreateMergedIf()
260 builder.setInsertionPointToEnd(&new_if_op.then_branch().front()); in CreateMergedIf()
[all …]
|
| D | tpu_extract_outside_compilation.cc | 135 auto& then_branch = host_side_if.then_branch(); in CloneEmptyIfWithPredicate() local
136 then_branch.push_back(new Block); in CloneEmptyIfWithPredicate()
137 builder.setInsertionPointToEnd(&then_branch.front()); in CloneEmptyIfWithPredicate()
365 MoveOpsToHost(tpu_cluster, &if_op.then_branch().front(), in DecomposeControlFlow()
366 host_if.then_branch().front().getTerminator(), in DecomposeControlFlow()
|
| D | region_control_flow_to_functional.cc | 292 const TrivialTransformInfo tti(IsSingleCallRegion(if_region.then_branch()), in ConvertIfOp()
305 CollectExternValues(if_region.then_branch(), if_region.else_branch()); in ConvertIfOp()
321 ExtractSingleBlockRegion(if_region.then_branch(), then_name, extern_values, in ConvertIfOp()
|
| D | collection_ops_util.cc | 183 auto then_branch = if_op.then_function(); in GetElementTypeFromAccess() local
185 assert(then_branch && else_branch); in GetElementTypeFromAccess()
187 then_branch.getArgument(use.getOperandNumber() - 1), module, in GetElementTypeFromAccess()
|
| D | tensor_list_ops_decomposition.cc | 351 Region& then_branch = if_op.then_branch(); in HandleIfRegionOp() local
354 &then_branch.front(), module, buffer_to_size, in HandleIfRegionOp()
363 then_branch.front(), *buffer_to_size); in HandleIfRegionOp()
371 if_op.getLoc(), then_branch.front().getTerminator()->getOperandTypes(), in HandleIfRegionOp()
378 new_op.then_branch().takeBody(if_op.then_branch()); in HandleIfRegionOp()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | If.pbtxt | 32 name: "then_branch"
70 name: "then_branch"
107 name: "then_branch"
144 name: "then_branch"
182 name: "then_branch"
|
| D | StatelessIf.pbtxt | 30 name: "then_branch"
67 name: "then_branch"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | If.pbtxt | 32 name: "then_branch"
70 name: "then_branch"
107 name: "then_branch"
144 name: "then_branch"
182 name: "then_branch"
|
| D | StatelessIf.pbtxt | 30 name: "then_branch"
67 name: "then_branch"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | functional-control-flow-to-regions.mlir | 12 then_branch = @testIf1Then, else_branch = @testIf1Else, is_stateless = false,
28 // CHECK-NOT: then_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
254 func private @then_branch() -> ()
260 …"tf.If"(%arg0) {then_branch = @then_branch, else_branch = @else_branch, is_stateless = false, devi…
|
| D | region-control-flow-to-functional.mlir | 13 // CHECK-SAME: then_branch = @test_then_name
36 // CHECK-SAME: then_branch = @tf.IfRegion_then
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/compiler/tf2xla/ |
| D | functionalize_cond_test.cc | 83 StateMap::CondId then_branch; in TEST_F() local
87 then_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()
106 auto t = JoinCondStatesNonMerge(then_branch, joined); in TEST_F()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_If.pbtxt | 26 name: "then_branch"
36 types are the same as what then_branch returns.
39 summary: "output = cond ? then_branch(input) : else_branch(input)"
|
| D | api_def_StatelessIf.pbtxt | 29 name: "then_branch"
39 types are the same as what then_branch returns.
42 summary: "output = cond ? then_branch(input) : else_branch(input)"
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | auto_control_deps_test.py | 232 def then_branch(): function
239 constant_op.constant(True), then_branch, else_branch)
247 def then_branch(): function
254 constant_op.constant(False), then_branch, else_branch)
352 def then_branch(): function
359 constant_op.constant(True), then_branch, else_branch)
422 def then_branch(): function
430 constant_op.constant(True), then_branch, else_branch)
438 def then_branch(): function
446 constant_op.constant(False), then_branch, else_branch)
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| D | if_op.mlir | 4 …}, %{{.*}}, %{{.*}}) {else_branch = @cond_false, is_stateless = false, then_branch = @cond_true} :…
6 …%1 = "tf.If"(%0, %arg0, %arg1) {else_branch = @cond_false, then_branch = @cond_true, is_stateless …
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | optimize_functional_ops.mlir | 12 …%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…
79 then_branch = @_functionalize_if_then_branch_00} :
112 then_branch = @_functionalize_if_then_branch_01} :
148 then_branch = @_functionalize_if_then_branch_02} :
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | graph-function-defs.pbtxt | 5 # CHECK-SAME: then_branch = @
6 # CHECK-SAME: then_branch.how_many = 32
7 # CHECK-SAME: then_branch.ping = "ack"
232 key: "then_branch"
|
| D | functional-if-ops.pbtxt | 62 key: "then_branch"
110 key: "then_branch"
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/end2end/ |
| D | if_op.pbtxt | 126 key: "then_branch"
174 key: "then_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/compiler/glsl/ |
| D | ir_builder.cpp | 621 ir_instruction *then_branch) in if_tree() argument
623 assert(then_branch != NULL); in if_tree()
628 result->then_instructions.push_tail(then_branch); in if_tree()
634 ir_instruction *then_branch, in if_tree() argument
637 assert(then_branch != NULL); in if_tree()
643 result->then_instructions.push_tail(then_branch); in if_tree()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | functional_ops.py | 819 def If(cond, inputs, then_branch, else_branch, name=None): argument
843 if isinstance(then_branch, function._DefinedFunction):
844 tlist = [_.type for _ in then_branch.definition.signature.output_arg]
846 cond, inputs, tlist, then_branch, else_branch, name=name)
849 then_out = then_branch.structured_outputs
856 tlist = nest.flatten(then_branch.output_dtypes)
858 cond, inputs, tlist, then_branch, else_branch, name=name)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/ |
| D | functional-if-ops.mlir | 6 …uts_2, %arg2, %arg3) {else_branch = @cond_false, is_stateless = false, then_branch = @cond_true} :…
7 …puts_2, %arg2, %arg3) {else_branch = @cond_false, is_stateless = true, then_branch = @cond_true} :…
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | if_conditional.hlotxt | 5 %then_branch {
45 …(f32[]) conditional(%compare.4, %tuple.5, %tuple.5), true_computation=%then_branch, false_computat…
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_control_flow.cc | 262 Value then_input = TupleImplicitInputs(op.then_branch(), loc, &builder); in LowerIfRegion()
263 ReplaceTerminator(&op.then_branch().front(), /*extra_results=*/{}, &builder); in LowerIfRegion()
275 if_op.true_branch().takeBody(op.then_branch()); in LowerIfRegion()
|