| /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/tf2xla/kernels/ |
| D | case_op.cc | 55 int32 branch_index = branch_index_literal.Get<int32>({}); in GetPrunedBranchesAndIndex() local
83 xla::XlaOp branch_index; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | conditional_thunk.cc | 76 int32 branch_index = -1; in ExecuteOnStream() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond.cc | 444 int branch_index = static_cast<int>(branch); in BuildArgumentNodes() local
456 int branch_index = e->src_output(); in BuildArgumentNodes() local
567 int branch_index = static_cast<int>(branch); in ExtractBodies() local
721 int branch_index = static_cast<int>(branch); in ExtractBodies() local
733 int branch_index = static_cast<int>(find_branch(e)); in ExtractBodies() local
756 int branch_index = static_cast<int>(branch); in BuildIfNode() local
934 int branch_index = static_cast<int>(branch); in BuildAndReplace() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | cond_v2.py | 983 def indexed_case(branch_index, argument
1119 def _build_case(branch_index, argument
|
| D | control_flow_ops.py | 3264 branch_index): argument
3326 branch_index, argument
3581 def switch_case(branch_index, argument
|
| /external/llvm-project/lldb/source/Target/ |
| D | ThreadPlanStepRange.cpp | 331 uint32_t branch_index = instructions->GetIndexOfNextBranchInstruction( in SetNextBranchBreakpoint() local
|
| D | Process.cpp | 5958 uint32_t branch_index = insn_list->GetIndexOfNextBranchInstruction( in AdvanceAddressToNextBranchInstruction() local
|
| /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 | 100 int branch_index = 0; in TryRemoveConditional() local
|
| D | dynamic_dimension_inference.cc | 1266 for (int64 branch_index = 0; branch_index < hlo->branch_count(); in HandleConditional() local
|
| D | shape_inference.cc | 2784 const Shape& branch_index, in InferConditionalShape()
|
| D | hlo_evaluator.cc | 1976 int branch_index; in HandleConditional() local
|
| D | hlo_instruction.cc | 1216 const Shape& shape, HloInstruction* branch_index, in CreateConditional()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | compilability_check_util_test.cc | 390 auto branch_index = ops::Placeholder(root.WithOpName("pred"), DT_INT32); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | choose_fastest_branch_dataset_op.cc | 498 Status MakeCurrentIterator(IteratorContext* ctx, int64 branch_index, in MakeCurrentIterator()
|
| /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 | 2543 HloInstruction* branch_index; in HandleInfeed() local
2946 HloInstruction* branch_index; in HandleOutfeed() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2362 XlaOp branch_index, in Conditional()
2378 XlaOp branch_index, in ConditionalImpl()
4862 XlaOp Conditional(const XlaOp branch_index, in Conditional()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 762 static LogicalResult VerifyCaseOpBase(Operation *op, Value branch_index) { in VerifyCaseOpBase()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2634 auto branch_index = conditional->operand(0); in HandleConditional() local
|