Searched refs:dfs_stack (Results 1 – 3 of 3) sorted by relevance
50 InlinedVector<DFSFrame, 8> dfs_stack; in CreateLoopInvariantCopy() local51 dfs_stack.push_back({to_hoist, 0}); in CreateLoopInvariantCopy()57 DFSFrame* frame = &dfs_stack.back(); in CreateLoopInvariantCopy()86 dfs_stack.pop_back(); in CreateLoopInvariantCopy()97 dfs_stack.push_back({next_operand, 0}); in CreateLoopInvariantCopy()98 } while (!dfs_stack.empty()); in CreateLoopInvariantCopy()
320 std::vector<HloInstruction*> dfs_stack; in ComputeInstructionPostOrder() local321 dfs_stack.push_back(root); in ComputeInstructionPostOrder()322 while (!dfs_stack.empty()) { in ComputeInstructionPostOrder()323 const auto current = dfs_stack.back(); in ComputeInstructionPostOrder()328 dfs_stack.pop_back(); in ComputeInstructionPostOrder()333 dfs_stack.pop_back(); in ComputeInstructionPostOrder()360 dfs_stack.emplace_back(cinst); in ComputeInstructionPostOrder()363 dfs_stack.emplace_back(inst); in ComputeInstructionPostOrder()965 std::vector<HloInstruction*> dfs_stack; in CloneWithReplacements() local970 dfs_stack.push_back(new_instr); in CloneWithReplacements()[all …]
2687 inline bool PushDFSChild(Visitor* visitor, DFSStack* dfs_stack, in PushDFSChild() argument2701 dfs_stack->push_back(std::make_pair(id, child)); in PushDFSChild()2721 DFSStack dfs_stack; in PostOrderDFS() local2722 dfs_stack.emplace_back(root->unique_id(), root); in PostOrderDFS()2725 DCHECK(!dfs_stack.empty()); in PostOrderDFS()2727 int current_id = dfs_stack.back().first; in PostOrderDFS()2728 HloInstruction* current_node = dfs_stack.back().second; in PostOrderDFS()2734 dfs_stack.pop_back(); in PostOrderDFS()2741 dfs_stack.pop_back(); in PostOrderDFS()2753 const size_t old_dfs_stack_size = dfs_stack.size(); in PostOrderDFS()[all …]