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()
278 VisitState GetVisitState(int id) { in GetVisitState() function285 VisitState GetVisitState(const HloInstruction& instruction);310 return GetVisitState(instruction) == kVisiting; in IsVisiting()315 return GetVisitState(instruction) == kVisited; in DidVisit()320 return GetVisitState(instruction) == kNotVisited; in NotVisited()
2692 switch (visitor->GetVisitState(id)) { in PushDFSChild()2732 visitor->GetVisitState(current_id); in PostOrderDFS()