Searched refs:VisitState (Results 1 – 5 of 5) sorted by relevance
81 struct VisitState { struct82 explicit VisitState(unsigned N) in VisitState() argument88 ~VisitState() { in ~VisitState() argument104 VisitState *NextState; argument108 VisitState *FirstVisitState;110 VisitState *allocateVisitState();111 void returnVisitState(VisitState *State);
239 enum VisitState { enum245 VisitState GetVisitState(int id) { return visit_state_.GetState(id); } in GetVisitState()246 VisitState GetVisitState(const HloInstruction& instruction);257 void SetVisitState(int id, VisitState state) { in SetVisitState()314 VisitState GetState(uint64 id) { in GetState()317 return VisitState::kNotVisited; in GetState()319 static_assert(static_cast<int>(VisitState::kVisited) < 3, in GetState()323 return static_cast<VisitState>((w >> shift) & 0x3); in GetState()325 void SetState(uint64 id, VisitState state) { in SetState()
42 typename DfsHloVisitorBase<HloInstructionPtr>::VisitState53 visit_state_.SetState(instruction.unique_id(), VisitState::kVisiting); in SetVisiting()61 visit_state_.SetState(instruction.unique_id(), VisitState::kVisited); in SetVisited()
2741 typename Visitor::VisitState visit_state = in PostOrderDFS()
255 ModuleManager::VisitState *ModuleManager::allocateVisitState() { in allocateVisitState()258 VisitState *Result = FirstVisitState; in allocateVisitState()265 return new VisitState(size()); in allocateVisitState()268 void ModuleManager::returnVisitState(VisitState *State) { in returnVisitState()357 VisitState *State = allocateVisitState(); in visit()