Searched refs:GetVisitState (Results 1 – 3 of 3) sorted by relevance
43 DfsHloVisitorBase<HloInstructionPtr>::GetVisitState( in GetVisitState() function in xla::DfsHloVisitorBase45 return GetVisitState(instruction.unique_id()); in GetVisitState()
245 VisitState GetVisitState(int id) { return visit_state_.GetState(id); } in GetVisitState() function246 VisitState GetVisitState(const HloInstruction& instruction);273 return GetVisitState(instruction) == kVisiting; in IsVisiting()278 return GetVisitState(instruction) == kVisited; in DidVisit()283 return GetVisitState(instruction) == kNotVisited; in NotVisited()
2702 switch (visitor->GetVisitState(id)) { in PushDFSChild()2742 visitor->GetVisitState(current_id); in PostOrderDFS()