| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | auto_control_deps_test.py | 321 def then_branch(): function
336 def then_branch(): function
441 def then_branch(): function
511 def then_branch(): function
527 def then_branch(): function
638 def then_branch(): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | dtensor_op_spmd_expander.cc | 338 auto& then_branch = send_if.then_branch(); in ExpandOp() local
475 auto& then_branch = recv_if.then_branch(); in ExpandOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
| H A D | cond_v2_test.py | 269 def then_branch(): function
1289 def then_branch(): function
1313 def then_branch(): function
|
| /aosp_15_r20/external/mesa3d/src/compiler/glsl/ |
| H A D | ir_builder.cpp | 614 ir_instruction *then_branch) in if_tree()
627 ir_instruction *then_branch, in if_tree()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | functionalize_cond_test.cc | 82 StateMap::CondId then_branch; in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | decompose_reduce_dataset.cc | 173 auto& then_branch = dataset_if.then_branch(); in CreateOptionalDatasetIf() local
|
| H A D | collection_ops_util.cc | 183 auto then_branch = if_op.then_function(); in GetElementTypeFromAccess() local
|
| H A D | tpu_extract_outside_compilation.cc | 207 auto& then_branch = host_side_if.then_branch(); in CloneEmptyIfWithPredicate() local
|
| H A D | tensor_list_ops_decomposition.cc | 338 Region& then_branch = if_op.then_branch(); in HandleIfRegionOp() local
|
| H A D | resource_op_lifting.cc | 1183 auto then_branch = if_op.then_function(); in HoistForControlFlow() local
|
| H A D | tensor_array_ops_decomposition.cc | 644 auto then_branch = if_op.then_function(); in HandleIfOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | functional_ops.py | 818 def If(cond, inputs, then_branch, else_branch, name=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | merge_clusters.cc | 310 auto& then_branch = host_side_if.then_branch(); in CloneEmptyIfWithPredicate() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| H A D | fusion_utils.cc | 397 NameAttrList then_branch; in LazyConjunctionNodes() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/ |
| H A D | export.cpp | 1010 auto then_branch = node_proto->add_attribute(); in EncodeNode() local
|
| /aosp_15_r20/external/mesa3d/src/panfrost/midgard/ |
| H A D | midgard_compile.c | 2757 midgard_instruction *then_branch = mir_last_in_block(ctx->current_block); in emit_if() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| H A D | tf_to_tfrt.cc | 1097 mlir::FlatSymbolRefAttr then_branch = op.then_branchAttr(); in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | extract_outside_compilation_pass.cc | 1903 NameAttrList then_branch, else_branch; in ExtractOutsideCompilationForIfNode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | benchmarks_test.py | 1391 def then_branch(): function
|
| /aosp_15_r20/external/mesa3d/src/panfrost/compiler/ |
| H A D | bifrost_compile.c | 4100 bi_instr *then_branch = in emit_if() local
|