| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | functionalize_cond.cc | 442 int branch_index = static_cast<int>(branch); in BuildArgumentNodes() local
454 int branch_index = e->src_output(); in BuildArgumentNodes() local
565 int branch_index = static_cast<int>(branch); in ExtractBodies() local
719 int branch_index = static_cast<int>(branch); in ExtractBodies() local
731 int branch_index = static_cast<int>(find_branch(e)); in ExtractBodies() local
754 int branch_index = static_cast<int>(branch); in BuildIfNode() local
929 int branch_index = static_cast<int>(branch); in BuildAndReplace() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | lower_case_op_test.cc | 79 auto branch_index = in TEST() local
226 auto branch_index = in TEST() local
348 auto branch_index = in TEST() local
|
| H A D | lower_case_op.cc | 121 Node* branch_index; in CreatePivotNodes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | case_op.cc | 60 int32_t branch_index = branch_index_literal.Get<int32>({}); in GetPrunedBranchesAndIndex() local
88 xla::XlaOp branch_index; in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | cond_v2.py | 1024 def indexed_case(branch_index, argument
1160 def _build_case(branch_index, argument
|
| H A D | control_flow_ops.py | 3246 branch_index): argument
3309 branch_index, argument
3564 def switch_case(branch_index, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | save_restore_spmd_expander.cc | 369 mlir::Value branch_index = DeviceIdToLocalBranchIndex( in ConditionalSave() local
642 mlir::Value branch_index = DeviceIdToLocalBranchIndex( in ExpandRestoreV2OpHelper() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | case_test.py | 32 def switch_case_test(branch_index): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | conditional_thunk.cc | 51 int32_t branch_index = -1; in ExecuteOnStream() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | conditional_test.cc | 199 XlaOp branch_index; in XLA_TEST_P() local
242 XlaOp branch_index; in XLA_TEST_P() local
416 XlaOp branch_index; in XLA_TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | dynamic_shaped_ops.cc | 167 XlaBuilder* builder, XlaOp branch_index, in DynamicConditional()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | conditional_simplifier.cc | 475 int branch_index = 0; in TryRemoveConditional() local
|
| H A D | dynamic_dimension_inference.cc | 1441 for (int64_t branch_index = 0; branch_index < hlo->branch_count(); in HandleConditional() local
1545 for (int64_t branch_index = 0; branch_index < hlo->branch_count(); in HandleConditional() local
|
| H A D | shape_inference.cc | 2901 const Shape& branch_index, in InferConditionalShape()
|
| H A D | hlo_evaluator.cc | 3182 int branch_index; in HandleConditional() local
|
| H A D | hlo_instruction.cc | 1490 const Shape& shape, HloInstruction* branch_index, in CreateConditional()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | value_inference.cc | 577 int64_t branch_index = 0; in AnalyzeConstantValueFallback() local
1275 int64_t branch_index = 0; in AnalyzeIsDynamic() local
|
| H A D | xla_builder.cc | 2517 XlaOp branch_index, in Conditional()
2533 XlaOp branch_index, in ConditionalImpl()
4886 XlaOp Conditional(const XlaOp branch_index, in Conditional()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | choose_fastest_branch_dataset_op.cc | 498 Status MakeCurrentIterator(IteratorContext* ctx, int64_t branch_index, in MakeCurrentIterator()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | compilability_check_util_test.cc | 420 auto branch_index = ops::Placeholder(root.WithOpName("pred"), DT_INT32); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | functional_ops.cc | 269 const Tensor& branch_index = ctx->input(0); in ComputeAsync() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/ |
| H A D | ops.cc | 1285 } else if (auto branch_index = GetStaticallyKnownCaseBranch(operands[0])) { in GetCaseLikeRegionOpSuccessorRegions() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | spmd_partitioner.cc | 3298 HloInstruction* branch_index; in HandleInfeed() local
3727 HloInstruction* branch_index; in HandleOutfeed() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | ir_emitter.cc | 2813 auto branch_index = conditional->operand(0); in HandleConditional() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_ops_a_m.cc | 809 static LogicalResult VerifyCaseOpBase(Operation *op, Value branch_index) { in VerifyCaseOpBase()
|