Home
last modified time | relevance | path

Searched refs:HloInstructionSequence (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_schedule.h34 class HloInstructionSequence {
36 HloInstructionSequence() = default;
37 explicit HloInstructionSequence( in HloInstructionSequence() function
91 const HloInstructionSequence& sequence(
96 HloInstructionSequence& GetOrCreateSequence(
103 HloInstructionSequence sequence);
107 const absl::flat_hash_map<int64, HloInstructionSequence>& sequences() const { in sequences()
151 absl::flat_hash_map<int64, HloInstructionSequence> sequences_;
Dhlo_memory_scheduler.h38 typedef std::function<StatusOr<HloInstructionSequence>(
45 StatusOr<HloInstructionSequence> ListMemoryScheduler(
53 StatusOr<HloInstructionSequence> DFSMemoryScheduler(
61 StatusOr<HloInstructionSequence> PostOrderMemoryScheduler(
71 StatusOr<HloInstructionSequence> DefaultMemoryScheduler(
87 StatusOr<HloInstructionSequence> ScheduleComputation(
Dhlo_memory_scheduler.cc77 static StatusOr<HloInstructionSequence> Run( in Run()
260 HloInstructionSequence CreateSchedule() { in CreateSchedule()
261 HloInstructionSequence schedule; in CreateSchedule()
401 StatusOr<HloInstructionSequence> ScheduleComputationHelper( in ScheduleComputationHelper()
419 StatusOr<HloInstructionSequence> DFSMemoryScheduler( in DFSMemoryScheduler()
470 HloInstructionSequence sequence; in DFSMemoryScheduler()
490 StatusOr<HloInstructionSequence> ListMemoryScheduler( in ListMemoryScheduler()
500 StatusOr<HloInstructionSequence> PostOrderMemoryScheduler( in PostOrderMemoryScheduler()
506 return HloInstructionSequence(computation->MakeInstructionPostOrder()); in PostOrderMemoryScheduler()
509 StatusOr<HloInstructionSequence> DefaultMemoryScheduler( in DefaultMemoryScheduler()
[all …]
Dhlo_schedule.cc54 HloInstructionSequence& sequence = in CreateFromProto()
73 const HloInstructionSequence& sequence = id_sequence.second; in ToProto()
86 set_sequence(computation, HloInstructionSequence(sequence)); in set_sequence()
90 HloInstructionSequence sequence) { in set_sequence()
95 HloInstructionSequence& HloSchedule::GetOrCreateSequence( in GetOrCreateSequence()
107 const HloInstructionSequence& HloSchedule::sequence( in sequence()
158 HloInstructionSequence new_sequence; in UpdateComputationSchedule()
Dhlo_ordering.h68 virtual const HloInstructionSequence* SequentialOrder(
100 const HloInstructionSequence* SequentialOrder( in SequentialOrder()
189 const HloInstructionSequence* SequentialOrder(
Dheap_simulator.h100 const HloComputation& computation, const HloInstructionSequence& sequence,
129 const HloInstructionSequence& instruction_sequence,
148 const HloInstructionSequence& instruction_sequence,
Dhlo_rematerialization.h91 const HloInstructionSequence& order) const;
Dheap_simulator.cc58 const HloComputation& computation, const HloInstructionSequence& sequence, in MinimumMemoryForComputation()
80 const HloInstructionSequence& instruction_sequence = in Run()
90 const HloInstructionSequence& instruction_sequence, in Run()
106 const HloInstructionSequence& instruction_sequence, in RunComputation()
301 const HloInstructionSequence& called_sequence = in RunComputation()
Dhlo_rematerialization.cc142 explicit InstructionList(const HloInstructionSequence& order) { in InstructionList()
936 const HloInstructionSequence& order) const { in ComputePeakMemory()
1188 HloInstructionSequence& sequence = schedule->GetOrCreateSequence(computation); in RematerializeComputation()
Dhlo_ordering.cc380 const HloInstructionSequence* SequentialHloOrdering::SequentialOrder( in SequentialOrder()
Dbuffer_assignment.cc636 const HloInstructionSequence* sequence = in ComputeSummaryStats()
1179 const HloInstructionSequence* instruction_sequence = in AssignBuffersWithSequentialOrdering()
1214 const HloInstructionSequence* instruction_sequence = in AssignBuffersWithSequentialOrdering()
Dhlo_module.cc577 HloInstructionSequence& clone_sequence = in Clone()
Dheap_simulator_test.cc277 HloInstructionSequence(instruction_sequence), in HeapSimulatorTracker()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_hlo_schedule.cc45 const HloInstructionSequence* SequentialOrder( in SequentialOrder()
54 std::unique_ptr<HloInstructionSequence> entry_sequence_;
64 absl::make_unique<HloInstructionSequence>(thunk_launch_order); in GpuHloOrdering()
203 HloInstructionSequence sequence, in Build()