Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dheap_simulator.cc106 const HloInstructionSequence& instruction_sequence = in Run() local
112 instruction_sequence, alias_analysis, in Run()
121 const HloInstructionSequence& instruction_sequence, in Run() argument
129 schedule.set_sequence(&computation, instruction_sequence); in Run()
133 TF_RETURN_IF_ERROR(heap.RunComputation(computation, instruction_sequence, in Run()
142 const HloInstructionSequence& instruction_sequence, in Run() argument
151 TF_RETURN_IF_ERROR(heap.RunComputation(computation, instruction_sequence, in Run()
160 const HloInstructionSequence& instruction_sequence, in RunComputation() argument
H A Dbuffer_assignment.cc1350 const HloInstructionSequence* instruction_sequence = in AssignBuffersForComputations() local
1352 const bool has_sequential_order = instruction_sequence != nullptr; in AssignBuffersForComputations()
1361 schedule.set_sequence(computation, *instruction_sequence); in AssignBuffersForComputations()
1500 const HloInstructionSequence* instruction_sequence = in AssignBuffersWithSequentialOrdering() local
1502 CHECK(instruction_sequence != nullptr) << computation->name(); in AssignBuffersWithSequentialOrdering()
1503 schedule.set_sequence(computation, *instruction_sequence); in AssignBuffersWithSequentialOrdering()
1532 const HloInstructionSequence* instruction_sequence = in AssignBuffersWithSequentialOrdering() local
1534 CHECK(instruction_sequence != nullptr) << computation->name(); in AssignBuffersWithSequentialOrdering()
1545 *instruction_sequence, in AssignBuffersWithSequentialOrdering()
1724 const HloInstructionSequence* instruction_sequence = in CreateAssignment() local
[all …]
H A Dheap_simulator.h163 const HloInstructionSequence& instruction_sequence,
175 const HloInstructionSequence& instruction_sequence,
192 const HloInstructionSequence& instruction_sequence,
H A Dhlo_live_range.cc90 const HloInstructionSequence& instruction_sequence = it->second; in FlattenSchedule() local
91 for (HloInstruction* instruction : instruction_sequence.instructions()) { in FlattenSchedule()
H A Dheap_simulator_test.cc276 const std::vector<HloInstruction*>& instruction_sequence, in HeapSimulatorTracker() argument
280 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()
H A Dmemory_space_assignment.cc1444 const auto& instruction_sequence = in Finish() local
1446 for (int i = 0; i < instruction_sequence.size(); ++i) { in Finish()
1447 VLOG(3) << " " << i << ": " << instruction_sequence[i]->parent()->name() in Finish()
1448 << " " << instruction_sequence[i]->name(); in Finish()
2253 const auto& instruction_sequence = in AllocateReservedScopedAllocations() local
2256 for (int i = 0; i < instruction_sequence.size(); ++i) { in AllocateReservedScopedAllocations()
2257 const HloInstruction* instruction = instruction_sequence[i]; in AllocateReservedScopedAllocations()
2282 HloPosition{instruction_sequence[i], {}}, MemorySpace::kAlternate, in AllocateReservedScopedAllocations()
3568 const auto& instruction_sequence = in ComputeEstimatedElapsedTime() local
3571 for (const HloInstruction* instruction : instruction_sequence) { in ComputeEstimatedElapsedTime()
H A Dbuffer_assignment_test.cc152 HloModule* module, absl::Span<HloInstruction* const> instruction_sequence, in RunBufferAssignmentWithInstructionSequence() argument
155 schedule.set_sequence(module->entry_computation(), instruction_sequence); in RunBufferAssignmentWithInstructionSequence()