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.cc1307 const HloInstructionSequence* instruction_sequence = in AssignBuffersForComputations() local
1309 const bool has_sequential_order = instruction_sequence != nullptr; in AssignBuffersForComputations()
1318 schedule.set_sequence(computation, *instruction_sequence); in AssignBuffersForComputations()
1457 const HloInstructionSequence* instruction_sequence = in AssignBuffersWithSequentialOrdering() local
1459 CHECK(instruction_sequence != nullptr) << computation->name(); in AssignBuffersWithSequentialOrdering()
1460 schedule.set_sequence(computation, *instruction_sequence); in AssignBuffersWithSequentialOrdering()
1489 const HloInstructionSequence* instruction_sequence = in AssignBuffersWithSequentialOrdering() local
1491 CHECK(instruction_sequence != nullptr) << computation->name(); in AssignBuffersWithSequentialOrdering()
1502 *instruction_sequence, in AssignBuffersWithSequentialOrdering()
1649 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.cc99 const HloInstructionSequence& instruction_sequence = in FlattenSchedule() local
102 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.cc1150 const auto& instruction_sequence = in Finish() local
1152 for (int i = 0; i < instruction_sequence.size(); ++i) { in Finish()
1153 VLOG(3) << " " << i << ": " << instruction_sequence[i]->parent()->name() in Finish()
1154 << " " << instruction_sequence[i]->name(); in Finish()
1782 const auto& instruction_sequence = in AllocateReservedScopedAllocations() local
1785 for (int i = 0; i < instruction_sequence.size(); ++i) { in AllocateReservedScopedAllocations()
1787 options_.reserved_scoped_memory_fn(instruction_sequence[i]); in AllocateReservedScopedAllocations()
1790 << instruction_sequence[i]->name() in AllocateReservedScopedAllocations()
1806 HloPosition{instruction_sequence[i], {}}, MemorySpace::kAlternate, in AllocateReservedScopedAllocations()
2999 const auto& instruction_sequence = in ComputeEstimatedElapsedTime() local
[all …]
Dbuffer_assignment_test.cc137 HloModule* module, absl::Span<HloInstruction* const> instruction_sequence, in RunBufferAssignmentWithInstructionSequence() argument
140 schedule.set_sequence(module->entry_computation(), instruction_sequence); in RunBufferAssignmentWithInstructionSequence()