Home
last modified time | relevance | path

Searched refs:kThenBranch (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dcond_builder.cc51 graph_->AddEdge(switch_pred, kThenBranch, switch_t_, 0); in CondBuilder()
58 graph_->AddEdge(switch_t_, 0, merge_pred, kThenBranch); in CondBuilder()
Dcond_builder.h39 enum Branch { kElseBranch = 0, kThenBranch = 1 }; enumerator
Ddistributed_tpu_rewrite_pass.cc2672 graph->AddEdge(switch_var, CondBuilder::kThenBranch, write_node, 0); in BuildVariableWrites()
2673 graph->AddEdge(switch_val, CondBuilder::kThenBranch, write_node, 1); in BuildVariableWrites()
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_if_op.cc37 enum Branch { kElseBranch = 0, kThenBranch = 1 }; enumerator
169 .Input(switch_pred, kThenBranch) in CreatePivotNodes()
203 then_call_builder_.Input(input, kThenBranch); in AddInput()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond.cc96 case BranchType::kThenBranch: 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()
550 TF_RET_CHECK(branch == BranchType::kThenBranch || 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()
1119 (diff[0].second == BranchType::kThenBranch || in JoinCondStatesMerge()
[all …]
Dfunctionalize_cond.h52 kThenBranch = 1, enumerator
Dfunctionalize_cond_test.cc86 ss.insert(std::make_pair(OutputTensor(pred, 0), BranchType::kThenBranch)); in TEST_F()