Searched refs:instruction_sequence (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | heap_simulator.cc | 106 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 D | buffer_assignment.cc | 1350 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 D | heap_simulator.h | 163 const HloInstructionSequence& instruction_sequence, 175 const HloInstructionSequence& instruction_sequence, 192 const HloInstructionSequence& instruction_sequence,
|
H A D | hlo_live_range.cc | 90 const HloInstructionSequence& instruction_sequence = it->second; in FlattenSchedule() local 91 for (HloInstruction* instruction : instruction_sequence.instructions()) { in FlattenSchedule()
|
H A D | heap_simulator_test.cc | 276 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 D | memory_space_assignment.cc | 1444 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 D | buffer_assignment_test.cc | 152 HloModule* module, absl::Span<HloInstruction* const> instruction_sequence, in RunBufferAssignmentWithInstructionSequence() argument 155 schedule.set_sequence(module->entry_computation(), instruction_sequence); in RunBufferAssignmentWithInstructionSequence()
|