Searched refs:kElseBranch (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | functionalize_cond.cc | 90 case BranchType::kElseBranch: in Branch_Name() 427 for (auto branch : {BranchType::kElseBranch, BranchType::kThenBranch}) { in BuildArgumentNodes() 464 for (auto branch : {BranchType::kElseBranch, BranchType::kThenBranch}) { in BuildArgumentNodes() 515 for (auto b : {BranchType::kElseBranch, BranchType::kThenBranch}) { in ExtractBodies() 535 branch == BranchType::kElseBranch) in ExtractBodies() 550 for (auto branch : {BranchType::kElseBranch, BranchType::kThenBranch}) { in ExtractBodies() 692 for (auto branch : {BranchType::kElseBranch, BranchType::kThenBranch}) { in ExtractBodies() 727 for (auto branch : {BranchType::kElseBranch, BranchType::kThenBranch}) { in BuildIfNode() 892 for (auto branch : {BranchType::kElseBranch, BranchType::kThenBranch}) { in BuildAndReplace() 1073 diff[0].second == BranchType::kElseBranch) && in JoinCondStatesMerge() [all …]
|
D | functionalize_cond_test.cc | 85 ss.insert(std::make_pair(OutputTensor(pred, 0), BranchType::kElseBranch)); in TEST_F()
|
D | functionalize_cond.h | 40 kElseBranch = 0, enumerator
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | lower_if_op.cc | 35 enum Branch { kElseBranch = 0, kThenBranch = 1 }; enumerator 132 .Input(switch_pred, kElseBranch) in CreatePivotNodes() 167 else_call_builder_.Input(input, kElseBranch); in AddInput()
|