Searched refs:cur_branch (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_constructor.cc | 224 void DFS(int cur_node, std::vector<int>* cur_branch,
1062 void GraphConstructor::DFS(int cur_node, std::vector<int>* cur_branch, in DFS() argument
1065 cur_branch->push_back(cur_node); in DFS()
1071 std::find(cur_branch->begin(), cur_branch->end(), next_node); in DFS()
1073 while (iter != cur_branch->end()) { in DFS()
1079 DFS(next_node, cur_branch, is_on_cur_branch, unvisited); in DFS()
1083 cur_branch->pop_back(); in DFS()
1098 std::vector<int> cur_branch; in PrintCycles() local
1104 DFS(cur_node, &cur_branch, &is_on_cur_branch, &unvisited); in PrintCycles()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | conditional_code_motion.cc | 475 HloComputation* cur_branch = conditional->branch_computation(branch); in ConvertSpecialMove() local
477 cur_branch->AddInstruction(HloInstruction::CreateTuple(new_operands)); in ConvertSpecialMove()
480 cur_branch->set_root_instruction(new_branch_root, true /*new shape*/); in ConvertSpecialMove()
481 TF_CHECK_OK(cur_branch->RemoveInstruction(old_root)); in ConvertSpecialMove()
490 cur_branch->root_instruction()->shape(), in ConvertSpecialMove()
|