Home
last modified time | relevance | path

Searched refs:kElseBranch (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dcond_builder.cc50 graph_->AddEdge(switch_pred, kElseBranch, switch_f_, 0); in CondBuilder()
57 graph_->AddEdge(switch_f_, 0, merge_pred, kElseBranch); in CondBuilder()
Dcond_builder.h39 enum Branch { kElseBranch = 0, kThenBranch = 1 }; enumerator
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_if_op.cc37 enum Branch { kElseBranch = 0, kThenBranch = 1 }; enumerator
163 .Input(switch_pred, kElseBranch) in CreatePivotNodes()
204 else_call_builder_.Input(input, kElseBranch); in AddInput()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond.cc94 case BranchType::kElseBranch: in Branch_Name()
443 for (auto branch : {BranchType::kElseBranch, BranchType::kThenBranch}) { in BuildArgumentNodes()
480 for (auto branch : {BranchType::kElseBranch, BranchType::kThenBranch}) { in BuildArgumentNodes()
531 for (auto b : {BranchType::kElseBranch, BranchType::kThenBranch}) { in ExtractBodies()
551 branch == BranchType::kElseBranch) in ExtractBodies()
566 for (auto branch : {BranchType::kElseBranch, BranchType::kThenBranch}) { in ExtractBodies()
720 for (auto branch : {BranchType::kElseBranch, BranchType::kThenBranch}) { in ExtractBodies()
755 for (auto branch : {BranchType::kElseBranch, BranchType::kThenBranch}) { in BuildIfNode()
933 for (auto branch : {BranchType::kElseBranch, BranchType::kThenBranch}) { in BuildAndReplace()
1120 diff[0].second == BranchType::kElseBranch) && in JoinCondStatesMerge()
[all …]
Dfunctionalize_cond.h51 kElseBranch = 0, enumerator
Dfunctionalize_cond_test.cc92 ss.insert(std::make_pair(OutputTensor(pred, 0), BranchType::kElseBranch)); in TEST_F()