| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_case_op_test.cc | 81 auto branch_index = in TEST() local 228 auto branch_index = in TEST() local 350 auto branch_index = in TEST() local
|
| D | lower_case_op.cc | 122 Node* branch_index; in CreatePivotNodes() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | conditional_thunk.cc | 65 int32 branch_index = -1; in ExecuteOnStream() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond.cc | 436 int branch_index = static_cast<int>(branch); in BuildArgumentNodes() local 448 int branch_index = e->src_output(); in BuildArgumentNodes() local 559 int branch_index = static_cast<int>(branch); in ExtractBodies() local 713 int branch_index = static_cast<int>(branch); in ExtractBodies() local 725 int branch_index = static_cast<int>(find_branch(e)); in ExtractBodies() local 748 int branch_index = static_cast<int>(branch); in BuildIfNode() local 922 int branch_index = static_cast<int>(branch); in BuildAndReplace() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | cond_v2.py | 943 def indexed_case(branch_index, branch_fns, name="indexed_case"): argument 1065 def _build_case(branch_index, branch_graphs, branch_inputs, name=None): argument
|
| D | control_flow_ops.py | 3200 branch_index): argument 3260 def _indexed_case_helper(branch_fns, default, branch_index, name): argument 3507 def switch_case(branch_index, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | conditional_simplifier.cc | 77 int branch_index = 0; in TryRemoveConditional() local
|
| D | shape_inference.cc | 2673 const Shape& branch_index, in InferConditionalShape()
|
| D | hlo_evaluator.cc | 1964 int branch_index; in HandleConditional() local
|
| D | hlo_instruction.cc | 1057 const Shape& shape, HloInstruction* branch_index, in CreateConditional()
|
| /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/core/kernels/data/experimental/ |
| D | choose_fastest_branch_dataset_op.cc | 480 Status MakeCurrentIterator(IteratorContext* ctx, int64 branch_index, in MakeCurrentIterator()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | functional_ops.cc | 235 const Tensor& branch_index = ctx->input(0); in ComputeAsync() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1853 XlaOp branch_index, in Conditional() 1869 XlaOp branch_index, in ConditionalImpl() 3477 XlaOp Conditional(const XlaOp branch_index, in Conditional()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2777 auto branch_index = conditional->operand(0); in HandleConditional() local
|