Searched refs:kThenBranch (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond.cc | 92 case BranchType::kThenBranch: 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()
534 TF_RET_CHECK(branch == BranchType::kThenBranch || 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()
1072 (diff[0].second == BranchType::kThenBranch || in JoinCondStatesMerge()
[all …]
|
| D | functionalize_cond_test.cc | 79 ss.insert(std::make_pair(OutputTensor(pred, 0), BranchType::kThenBranch)); in TEST_F()
|
| D | functionalize_cond.h | 41 kThenBranch = 1, enumerator
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_if_op.cc | 35 enum Branch { kElseBranch = 0, kThenBranch = 1 }; enumerator
137 .Input(switch_pred, kThenBranch) in CreatePivotNodes()
166 then_call_builder_.Input(input, kThenBranch); in AddInput()
|