Searched refs:VisitState (Results 1 – 9 of 9) 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);
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()341 TF_RET_CHECK(state.second == VisitState::kVisited); in VerifyComputations()
42 typename DfsHloVisitorBase<HloInstructionPtr>::VisitState53 visit_state_[instruction.unique_id()] = VisitState::kVisiting; in SetVisiting()61 visit_state_[instruction.unique_id()] = VisitState::kVisited; in SetVisited()
272 enum VisitState { enum278 VisitState GetVisitState(int id) { in GetVisitState()281 return VisitState::kNotVisited; in GetVisitState()285 VisitState GetVisitState(const HloInstruction& instruction);296 void SetVisitState(int id, VisitState state) { visit_state_[id] = state; } in SetVisitState()345 absl::flat_hash_map<int, VisitState> visit_state_;
62 enum VisitState { enum90 using VisitStates = absl::flat_hash_map<HloInstruction*, VisitState>;
448 enum VisitState { kVisiting, kVisited }; enum452 absl::flat_hash_map<HloInstruction*, VisitState>* visited) const;
319 absl::flat_hash_map<HloInstruction*, VisitState>* visited) const { in ComputeInstructionPostOrder()425 absl::flat_hash_map<HloInstruction*, VisitState> visited; in MakeInstructionPostOrder()963 absl::flat_hash_map<HloInstruction*, VisitState> visited; in CloneWithReplacements()
2731 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()