Searched refs:instruction_sequence (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | heap_simulator.cc | 80 const HloInstructionSequence& instruction_sequence = in Run() local
83 *entry_computation, instruction_sequence, points_to_analysis)); in Run()
90 const HloInstructionSequence& instruction_sequence, in Run() argument
97 TF_RETURN_IF_ERROR(heap.RunComputation(computation, instruction_sequence, in Run()
106 const HloInstructionSequence& instruction_sequence, in RunComputation() argument
140 instruction_sequence.instructions()) { in RunComputation()
174 instruction_sequence.instructions()) { in RunComputation()
|
| D | heap_simulator.h | 129 const HloInstructionSequence& instruction_sequence,
148 const HloInstructionSequence& instruction_sequence,
|
| D | buffer_assignment.cc | 1179 const HloInstructionSequence* instruction_sequence = in AssignBuffersWithSequentialOrdering() local
1181 CHECK(instruction_sequence != nullptr) << computation->name(); in AssignBuffersWithSequentialOrdering()
1182 schedule.set_sequence(computation, *instruction_sequence); in AssignBuffersWithSequentialOrdering()
1214 const HloInstructionSequence* instruction_sequence = in AssignBuffersWithSequentialOrdering() local
1216 CHECK(instruction_sequence != nullptr) << computation->name(); in AssignBuffersWithSequentialOrdering()
1229 *instruction_sequence, in AssignBuffersWithSequentialOrdering()
|
| D | heap_simulator_test.cc | 262 const std::vector<HloInstruction*>& instruction_sequence) { in HeapSimulatorTracker() argument
277 HloInstructionSequence(instruction_sequence), in HeapSimulatorTracker()
|
| D | buffer_assignment_test.cc | 141 HloModule* module, absl::Span<HloInstruction* const> instruction_sequence, in RunBufferAssignmentWithInstructionSequence() argument
144 schedule.set_sequence(module->entry_computation(), instruction_sequence); in RunBufferAssignmentWithInstructionSequence()
|