Home
last modified time | relevance | path

Searched refs:instruction_sequence (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dheap_simulator.cc103 const HloInstructionSequence& instruction_sequence = in Run() local
109 instruction_sequence, alias_analysis, in Run()
118 const HloInstructionSequence& instruction_sequence, in Run() argument
126 schedule.set_sequence(&computation, instruction_sequence); in Run()
130 TF_RETURN_IF_ERROR(heap.RunComputation(computation, instruction_sequence, in Run()
139 const HloInstructionSequence& instruction_sequence, in Run() argument
148 TF_RETURN_IF_ERROR(heap.RunComputation(computation, instruction_sequence, in Run()
157 const HloInstructionSequence& instruction_sequence, in RunComputation() argument
Dbuffer_assignment.cc1294 const HloInstructionSequence* instruction_sequence = in AssignBuffersForComputations() local
1296 const bool has_sequential_order = instruction_sequence != nullptr; in AssignBuffersForComputations()
1305 schedule.set_sequence(computation, *instruction_sequence); in AssignBuffersForComputations()
1444 const HloInstructionSequence* instruction_sequence = in AssignBuffersWithSequentialOrdering() local
1446 CHECK(instruction_sequence != nullptr) << computation->name(); in AssignBuffersWithSequentialOrdering()
1447 schedule.set_sequence(computation, *instruction_sequence); in AssignBuffersWithSequentialOrdering()
1476 const HloInstructionSequence* instruction_sequence = in AssignBuffersWithSequentialOrdering() local
1478 CHECK(instruction_sequence != nullptr) << computation->name(); in AssignBuffersWithSequentialOrdering()
1489 *instruction_sequence, in AssignBuffersWithSequentialOrdering()
1636 const HloInstructionSequence* instruction_sequence = in CreateAssignment() local
[all …]
Dheap_simulator.h157 const HloInstructionSequence& instruction_sequence,
169 const HloInstructionSequence& instruction_sequence,
186 const HloInstructionSequence& instruction_sequence,
Dhlo_live_range.cc97 const HloInstructionSequence& instruction_sequence = in FlattenSchedule() local
100 for (HloInstruction* instruction : instruction_sequence.instructions()) { in FlattenSchedule()
Dheap_simulator_test.cc277 const std::vector<HloInstruction*>& instruction_sequence, in HeapSimulatorTracker() argument
281 Init(instruction_sequence, can_share_buffer);
287 const std::vector<HloInstruction*>& instruction_sequence, in HeapSimulatorTracker() argument
293 Init(instruction_sequence, can_share_buffer);
379 void Init(const std::vector<HloInstruction*>& instruction_sequence, in Init() argument
395 HloInstructionSequence(instruction_sequence), in Init()
Dmemory_space_assignment.cc1108 const auto& instruction_sequence = in Finish() local
1110 for (int i = 0; i < instruction_sequence.size(); ++i) { in Finish()
1111 VLOG(3) << " " << i << ": " << instruction_sequence[i]->parent()->name() in Finish()
1112 << " " << instruction_sequence[i]->name(); in Finish()
Dbuffer_assignment_test.cc135 HloModule* module, absl::Span<HloInstruction* const> instruction_sequence, in RunBufferAssignmentWithInstructionSequence() argument
138 schedule.set_sequence(module->entry_computation(), instruction_sequence); in RunBufferAssignmentWithInstructionSequence()