Home
last modified time | relevance | path

Searched refs:kVisiting (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/utils/
Dtraversal.cc41 enum class NodeState { kNotVisited, kVisiting, kDone }; enumerator
85 if (state == NodeState::kVisiting) { in DfsTraversal()
93 state = NodeState::kVisiting; in DfsTraversal()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_group_util.cc278 if ((*visit_state)[hlo] == VisitState::kVisiting) { in VisitTopologicalOrder()
283 TF_RET_CHECK((*visit_state)[instruction] == VisitState::kVisiting); in VisitTopologicalOrder()
294 (*visit_state)[instruction] = VisitState::kVisiting; in VisitTopologicalOrder()
304 if ((*visit_state)[predecessor] == VisitState::kVisiting) { in VisitTopologicalOrder()
Ddfs_hlo_visitor.cc53 visit_state_[instruction.unique_id()] = VisitState::kVisiting; in SetVisiting()
Dhlo_module_group_util.h64 kVisiting, enumerator
Ddfs_hlo_visitor.h274 kVisiting = 1, enumerator
310 return GetVisitState(instruction) == kVisiting; in IsVisiting()
Dhlo_computation.cc332 CHECK_EQ(kVisiting, it->second); in ComputeInstructionPostOrder()
339 visited->insert({current, kVisiting}); in ComputeInstructionPostOrder()
980 CHECK_EQ(it->second, kVisiting); in CloneWithReplacements()
986 visited.insert({cur, kVisiting}); in CloneWithReplacements()
Dhlo_computation.h448 enum VisitState { kVisiting, kVisited }; enumerator
Dhlo_instruction.cc2693 case Visitor::kVisiting: in PushDFSChild()
2740 if (visit_state == Visitor::kVisiting) { in PostOrderDFS()
2751 visitor->SetVisitState(current_id, Visitor::kVisiting); in PostOrderDFS()