| /external/tensorflow/tensorflow/core/common_runtime/ |
| 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
|
| D | lower_case_op.cc | 121 Node* branch_index; in CreatePivotNodes() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | case_test.py | 36 def switch_case_test(branch_index): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | conditional_thunk.cc | 50 int32_t branch_index = -1; in ExecuteOnStream() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | case_op.cc | 59 int32_t branch_index = branch_index_literal.Get<int32>({}); in GetPrunedBranchesAndIndex() local
87 xla::XlaOp branch_index; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| 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
932 int branch_index = static_cast<int>(branch); in BuildAndReplace() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | cond_v2.py | 982 def indexed_case(branch_index, argument
1118 def _build_case(branch_index, argument
|
| D | control_flow_ops.py | 3287 branch_index): argument
3350 branch_index, argument
3605 def switch_case(branch_index, argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conditional_test.cc | 197 XlaOp branch_index; in XLA_TEST_P() local
240 XlaOp branch_index; in XLA_TEST_P() local
415 XlaOp branch_index; in XLA_TEST_P() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | conditional_simplifier.cc | 473 int branch_index = 0; in TryRemoveConditional() local
|
| D | dynamic_dimension_inference.cc | 1303 for (int64_t branch_index = 0; branch_index < hlo->branch_count(); in HandleConditional() local
1407 for (int64_t branch_index = 0; branch_index < hlo->branch_count(); in HandleConditional() local
|
| D | shape_inference.cc | 2898 const Shape& branch_index, in InferConditionalShape()
|
| D | hlo_evaluator.cc | 2040 int branch_index; in HandleConditional() local
|
| D | hlo_instruction.cc | 1393 const Shape& shape, HloInstruction* branch_index, in CreateConditional()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | choose_fastest_branch_dataset_op.cc | 498 Status MakeCurrentIterator(IteratorContext* ctx, int64_t branch_index, in MakeCurrentIterator()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | compilability_check_util_test.cc | 420 auto branch_index = ops::Placeholder(root.WithOpName("pred"), DT_INT32); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | value_inference.cc | 499 int64_t branch_index = 0; in AnalyzeConstantValueFallback() local
1120 int64_t branch_index = 0; in AnalyzeIsDynamic() local
|
| D | xla_builder.cc | 2390 XlaOp branch_index, in Conditional()
2406 XlaOp branch_index, in ConditionalImpl()
4724 XlaOp Conditional(const XlaOp branch_index, in Conditional()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | functional_ops.cc | 271 const Tensor& branch_index = ctx->input(0); in ComputeAsync() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2517 HloInstruction* branch_index; in HandleInfeed() local
2942 HloInstruction* branch_index; in HandleOutfeed() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 757 static LogicalResult VerifyCaseOpBase(Operation *op, Value branch_index) { in VerifyCaseOpBase()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2671 auto branch_index = conditional->operand(0); in HandleConditional() local
|