| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_schedule.h | 33 class HloInstructionSequence {
35 HloInstructionSequence() = default;
36 explicit HloInstructionSequence( in HloInstructionSequence() function
117 const HloInstructionSequence& sequence(
122 HloInstructionSequence& GetOrCreateSequence(
129 HloInstructionSequence sequence);
133 const absl::flat_hash_map<int64, HloInstructionSequence>& sequences() const { in sequences()
199 absl::flat_hash_map<int64, HloInstructionSequence> sequences_;
|
| D | hlo_memory_scheduler.h | 43 typedef std::function<StatusOr<HloInstructionSequence>(
63 StatusOr<HloInstructionSequence> ListMemoryScheduler(
73 StatusOr<HloInstructionSequence> DFSMemoryScheduler(
83 StatusOr<HloInstructionSequence> PostOrderMemoryScheduler(
96 StatusOr<HloInstructionSequence> DefaultMemoryScheduler(
121 StatusOr<HloInstructionSequence> ScheduleComputation(
|
| D | hlo_memory_scheduler.cc | 80 static StatusOr<HloInstructionSequence> Run( in Run()
272 HloInstructionSequence CreateSchedule() { in CreateSchedule()
273 HloInstructionSequence schedule; in CreateSchedule()
413 StatusOr<HloInstructionSequence> ScheduleComputationHelper( in ScheduleComputationHelper()
435 StatusOr<HloInstructionSequence> DFSMemoryScheduler( in DFSMemoryScheduler()
488 HloInstructionSequence sequence; in DFSMemoryScheduler()
527 HloInstructionSequence computation_sequence, in ComputationSchedulerToModuleScheduler()
542 StatusOr<HloInstructionSequence> ListMemoryScheduler( in ListMemoryScheduler()
550 TF_ASSIGN_OR_RETURN(HloInstructionSequence sequence, in ListMemoryScheduler()
562 StatusOr<HloInstructionSequence> PostOrderMemoryScheduler( in PostOrderMemoryScheduler()
[all …]
|
| D | hlo_schedule.cc | 54 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()
|
| D | root_instruction_sinker.cc | 29 HloInstructionSequence& sequence = in SinkTupleRoot()
44 HloInstructionSequence& sequence = in SinkNontupleRoot()
57 HloInstructionSequence& sequence = in Run()
|
| D | hlo_ordering.h | 89 virtual const HloInstructionSequence* SequentialOrder(
121 const HloInstructionSequence* SequentialOrder( in SequentialOrder()
210 const HloInstructionSequence* SequentialOrder(
|
| D | heap_simulator.h | 122 const HloComputation& computation, const HloInstructionSequence& sequence,
129 const HloComputation& computation, const HloInstructionSequence& sequence,
157 const HloInstructionSequence& instruction_sequence,
169 const HloInstructionSequence& instruction_sequence,
186 const HloInstructionSequence& instruction_sequence,
|
| D | hlo_live_range.h | 66 const HloInstructionSequence& flattened_instruction_sequence() const { in flattened_instruction_sequence()
203 HloInstructionSequence flattened_instruction_sequence_;
|
| D | heap_simulator.cc | 68 const HloComputation& computation, const HloInstructionSequence& sequence, in MinimumMemoryForComputation()
83 const HloComputation& computation, const HloInstructionSequence& sequence, in MinimumMemoryForComputation()
103 const HloInstructionSequence& instruction_sequence = in Run()
118 const HloInstructionSequence& instruction_sequence, in Run()
139 const HloInstructionSequence& instruction_sequence, in Run()
157 const HloInstructionSequence& instruction_sequence, in RunComputation()
|
| D | hlo_rematerialization.h | 127 const HloInstructionSequence& order) const;
|
| D | hlo_live_range.cc | 97 const HloInstructionSequence& instruction_sequence = in FlattenSchedule()
|
| D | buffer_assignment.cc | 728 const HloInstructionSequence* sequence = in ComputeSummaryStats()
1294 const HloInstructionSequence* instruction_sequence = in AssignBuffersForComputations()
1444 const HloInstructionSequence* instruction_sequence = in AssignBuffersWithSequentialOrdering()
1476 const HloInstructionSequence* instruction_sequence = in AssignBuffersWithSequentialOrdering()
1636 const HloInstructionSequence* instruction_sequence = in CreateAssignment()
|
| D | hlo_memory_scheduler_test.cc | 52 const HloInstructionSequence& sequence = schedule.sequence(computation); in PeakMemoryUseOfEntryComputation()
|
| D | hlo_ordering.cc | 517 const HloInstructionSequence* SequentialHloOrdering::SequentialOrder( in SequentialOrder()
|
| D | hlo_rematerialization.cc | 202 explicit InstructionList(const HloInstructionSequence& order) { in InstructionList()
1794 const HloInstructionSequence& order) const { in ComputePeakMemory()
1990 HloInstructionSequence& sequence = schedule->GetOrCreateSequence(computation); in RematerializeComputation()
|
| D | hlo_module.cc | 720 HloInstructionSequence& clone_sequence = in Clone()
|
| D | memory_space_assignment.h | 867 HloInstruction* new_instruction, HloInstructionSequence* new_sequence,
|
| D | memory_space_assignment.cc | 3109 HloInstruction* new_instruction, HloInstructionSequence* new_sequence, in EnsureInstructionAndOperandsInserted()
3193 HloInstructionSequence new_sequence; in FixSchedule()
|
| D | heap_simulator_test.cc | 395 HloInstructionSequence(instruction_sequence), in Init()
|
| D | memory_space_assignment_test.cc | 522 const HloInstructionSequence& sequence = in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_hlo_schedule.cc | 45 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()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1542 const xla::HloInstructionSequence* schedule = in HloToLhloModule()
|