Searched refs:VisitState (Results 1 – 10 of 10) sorted by relevance
| /external/clang/include/clang/Serialization/ |
| D | ModuleManager.h | 81 struct VisitState { struct
82 explicit VisitState(unsigned N) in VisitState() function
88 ~VisitState() { in ~VisitState() argument
104 VisitState *NextState; argument
108 VisitState *FirstVisitState;
110 VisitState *allocateVisitState();
111 void returnVisitState(VisitState *State);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_group_util.cc | 264 if ((*visit_state)[hlo] == VisitState::kVisited) { in VisitTopologicalOrder()
267 TF_RET_CHECK((*visit_state)[instruction] == VisitState::kVisited); in VisitTopologicalOrder()
273 if ((*visit_state)[hlo] == VisitState::kVisiting) { in VisitTopologicalOrder()
278 TF_RET_CHECK((*visit_state)[instruction] == VisitState::kVisiting); in VisitTopologicalOrder()
279 (*visit_state)[instruction] = VisitState::kVisited; in VisitTopologicalOrder()
287 TF_RET_CHECK((*visit_state)[instruction] == VisitState::kNotVisited) in VisitTopologicalOrder()
289 (*visit_state)[instruction] = VisitState::kVisiting; in VisitTopologicalOrder()
299 if ((*visit_state)[predecessor] == VisitState::kVisiting) { in VisitTopologicalOrder()
336 TF_RET_CHECK(state.second == VisitState::kVisited); in VerifyComputations()
|
| D | dfs_hlo_visitor.cc | 42 typename DfsHloVisitorBase<HloInstructionPtr>::VisitState
53 visit_state_[instruction.unique_id()] = VisitState::kVisiting; in SetVisiting()
61 visit_state_[instruction.unique_id()] = VisitState::kVisited; in SetVisited()
|
| D | dfs_hlo_visitor.h | 302 enum VisitState { enum
308 VisitState GetVisitState(int id) { in GetVisitState()
311 return VisitState::kNotVisited; in GetVisitState()
315 VisitState GetVisitState(const HloInstruction& instruction);
335 visit_state_ = absl::flat_hash_map<int, VisitState>{}; in DestroyVisitState()
338 void SetVisitState(int id, VisitState state) { visit_state_[id] = state; } in SetVisitState()
387 absl::flat_hash_map<int, VisitState> visit_state_;
|
| D | hlo_module_group_util.h | 62 enum VisitState { enum
90 using VisitStates = absl::flat_hash_map<HloInstruction*, VisitState>;
|
| D | hlo_computation.h | 696 enum VisitState { kVisiting, kVisited }; enum
700 absl::flat_hash_map<HloInstruction*, VisitState>& visited,
|
| D | hlo_computation.cc | 448 absl::flat_hash_map<HloInstruction*, VisitState>& visited, in ComputeInstructionPostOrder()
523 absl::flat_hash_map<HloInstruction*, VisitState> visited; in MakeInstructionPostOrder()
1230 absl::flat_hash_map<const HloInstruction*, VisitState> visited; in CloneWithReplacements()
|
| D | hlo_instruction.cc | 3765 typename Visitor::VisitState visit_state = in PostOrderDFS()
|
| /external/clang/lib/Serialization/ |
| D | ModuleManager.cpp | 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()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | value_inference.cc | 1491 enum VisitState { in PostOrderDFSVisit() enum
1500 PostorderDFSNodeType type, VisitState state, int64_t id) in PostOrderDFSVisit()
1509 VisitState state; in PostOrderDFSVisit()
|