Searched refs:dfs_stack (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_invariant_code_motion.cc | 50 InlinedVector<DFSFrame, 8> dfs_stack; in CreateLoopInvariantCopy() local
51 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()
|
| D | while_loop_expensive_invariant_code_motion.cc | 64 InlinedVector<DFSFrame, 8> dfs_stack; in CreateLoopInvariantCopy() local
65 dfs_stack.push_back({to_hoist, 0}); in CreateLoopInvariantCopy()
71 DFSFrame* frame = &dfs_stack.back(); in CreateLoopInvariantCopy()
98 dfs_stack.pop_back(); in CreateLoopInvariantCopy()
110 dfs_stack.push_back({next_operand, 0}); in CreateLoopInvariantCopy()
111 } while (!dfs_stack.empty()); in CreateLoopInvariantCopy()
|
| D | hlo_computation.cc | 385 std::vector<HloInstruction*> dfs_stack; in ComputeInstructionPostOrder() local
386 dfs_stack.push_back(root); in ComputeInstructionPostOrder()
387 while (!dfs_stack.empty()) { in ComputeInstructionPostOrder()
388 const auto current = dfs_stack.back(); in ComputeInstructionPostOrder()
396 dfs_stack.pop_back(); in ComputeInstructionPostOrder()
401 dfs_stack.pop_back(); in ComputeInstructionPostOrder()
428 dfs_stack.emplace_back(cinst); in ComputeInstructionPostOrder()
431 dfs_stack.emplace_back(inst); in ComputeInstructionPostOrder()
1062 std::vector<const HloInstruction*> dfs_stack; in CloneWithReplacements() local
1067 dfs_stack.push_back(new_instr); in CloneWithReplacements()
[all …]
|
| D | hlo_instruction.cc | 2564 void PrintCycle(const HloInstruction* child, DFSStack* dfs_stack) { in PrintCycle() argument
2569 while (!dfs_stack->empty() && dfs_stack->back().second != child) { in PrintCycle()
2570 subgraph.insert(dfs_stack->back().second); in PrintCycle()
2571 dfs_stack->pop_back(); in PrintCycle()
3290 inline bool PushDFSChild(Visitor* visitor, DFSStack* dfs_stack, in PushDFSChild() argument
3304 dfs_stack->push_back(std::make_pair(id, child)); in PushDFSChild()
3329 DFSStack dfs_stack; in PostOrderDFS() local
3330 dfs_stack.emplace_back(root->unique_id(), root); in PostOrderDFS()
3333 DCHECK(!dfs_stack.empty()); in PostOrderDFS()
3335 int current_id = dfs_stack.back().first; in PostOrderDFS()
[all …]
|