Home
last modified time | relevance | path

Searched refs:visit_state (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_group_util.cc244 VisitStates* visit_state, const VisitFunction& visit_function, in VisitTopologicalOrder() argument
269 if ((*visit_state)[hlo] == VisitState::kVisited) { in VisitTopologicalOrder()
272 TF_RET_CHECK((*visit_state)[instruction] == VisitState::kVisited); in VisitTopologicalOrder()
278 if ((*visit_state)[hlo] == VisitState::kVisiting) { in VisitTopologicalOrder()
283 TF_RET_CHECK((*visit_state)[instruction] == VisitState::kVisiting); in VisitTopologicalOrder()
284 (*visit_state)[instruction] = VisitState::kVisited; in VisitTopologicalOrder()
292 TF_RET_CHECK((*visit_state)[instruction] == VisitState::kNotVisited) in VisitTopologicalOrder()
294 (*visit_state)[instruction] = VisitState::kVisiting; in VisitTopologicalOrder()
304 if ((*visit_state)[predecessor] == VisitState::kVisiting) { in VisitTopologicalOrder()
Dhlo_module_group_util.h91 Status VisitTopologicalOrder(VisitStates* visit_state,
Dhlo_instruction.cc2731 typename Visitor::VisitState visit_state = in PostOrderDFS() local
2733 if (visit_state == Visitor::kVisited) { in PostOrderDFS()
2740 if (visit_state == Visitor::kVisiting) { in PostOrderDFS()