| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_schedule.cc | 33 /* static */ StatusOr<HloSchedule> HloSchedule::CreateFromProto( in CreateFromProto()
40 HloSchedule schedule(module); in CreateFromProto()
68 StatusOr<HloScheduleProto> HloSchedule::ToProto() const { in ToProto()
84 void HloSchedule::set_sequence(const HloComputation* computation, in set_sequence()
89 void HloSchedule::set_sequence(const HloComputation* computation, in set_sequence()
95 HloInstructionSequence& HloSchedule::GetOrCreateSequence( in GetOrCreateSequence()
107 const HloInstructionSequence& HloSchedule::sequence( in sequence()
112 Status HloSchedule::UpdateComputationSchedule( in UpdateComputationSchedule()
200 Status HloSchedule::Update() { in Update()
235 Status HloSchedule::Verify() const { in Verify()
[all …]
|
| D | hlo_schedule_test.cc | 57 HloSchedule schedule, in TEST_F()
91 HloSchedule schedule, in TEST_F()
140 HloSchedule schedule, in TEST_F()
184 HloSchedule schedule, in TEST_F()
245 HloSchedule schedule, in TEST_F()
314 HloSchedule schedule, in TEST_F()
|
| D | hlo_schedule.h | 107 class HloSchedule {
109 explicit HloSchedule(const HloModule* module) : module_(module) {} in HloSchedule() function
112 static StatusOr<HloSchedule> CreateFromProto(const HloModule* module,
202 std::ostream& operator<<(std::ostream& out, const HloSchedule& schedule);
|
| D | hlo_live_range.h | 44 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis,
100 explicit HloLiveRange(const HloSchedule& schedule, in HloLiveRange()
198 const HloSchedule& schedule_;
|
| D | hlo_live_range_test.cc | 42 void Analyze(const HloSchedule& schedule) { in Analyze()
97 HloSchedule schedule(module_.get()); in TEST_F()
126 HloSchedule schedule(module_.get()); in TEST_F()
163 HloSchedule schedule(module_.get()); in TEST_F()
202 HloSchedule schedule(module_.get()); in TEST_F()
242 HloSchedule schedule(module_.get()); in TEST_F()
323 HloSchedule schedule(module_.get()); in TEST_F()
|
| D | hlo_ordering.h | 205 explicit SequentialHloOrdering(const HloSchedule& schedule);
206 explicit SequentialHloOrdering(HloSchedule&& schedule);
221 const HloSchedule schedule_;
|
| D | hlo_memory_scheduler.cc | 521 int64* peak_memory) -> StatusOr<HloSchedule> { in ComputationSchedulerToModuleScheduler()
522 HloSchedule schedule(module); in ComputationSchedulerToModuleScheduler()
639 StatusOr<HloSchedule> DefaultModuleScheduler( in DefaultModuleScheduler()
653 HloSchedule list_sequence, in DefaultModuleScheduler()
661 TF_ASSIGN_OR_RETURN(HloSchedule dfs_sequence, in DefaultModuleScheduler()
669 HloSchedule post_order_sequence, in DefaultModuleScheduler()
696 StatusOr<HloSchedule> ScheduleModule( in ScheduleModule()
704 TF_ASSIGN_OR_RETURN(HloSchedule schedule, in ScheduleModule()
734 TF_ASSIGN_OR_RETURN(HloSchedule schedule, in Run()
741 HloSchedule schedule(module); in Run()
|
| D | hlo_memory_scheduler.h | 51 typedef std::function<StatusOr<HloSchedule>(
105 StatusOr<HloSchedule> DefaultModuleScheduler(
114 StatusOr<HloSchedule> ScheduleModule(
|
| D | heap_simulator.h | 116 const HloSchedule& schedule,
132 const HloSchedule* schedule);
145 const HloModule& module, const HloSchedule& schedule,
171 const BufferValue::SizeFunction& size_fn, const HloSchedule* schedule,
180 const Options& options, const HloSchedule* schedule = nullptr,
220 const HloSchedule* schedule_;
|
| D | hlo_memory_scheduler_test.cc | 49 const HloSchedule& schedule = module->schedule(); in PeakMemoryUseOfEntryComputation()
146 HloSchedule schedule, in TEST_F()
198 TF_ASSERT_OK_AND_ASSIGN(HloSchedule schedule, in TEST_F()
244 HloSchedule schedule, in TEST_F()
295 HloSchedule schedule, in TEST_F()
|
| D | hlo_module.h | 304 Status set_schedule(HloSchedule schedule);
313 const HloSchedule& schedule() const { return *schedule_; } in schedule()
314 HloSchedule& schedule() { return *schedule_; } in schedule()
413 absl::optional<HloSchedule> schedule_;
|
| D | memory_space_assignment_test.cc | 305 HloSchedule schedule(module.get()); in CreateEvictAndPrefetchModule()
412 HloSchedule schedule(module.get()); in TEST_P()
441 HloSchedule schedule(module.get()); in TEST_P()
496 HloSchedule schedule(module.get()); in TEST_P()
627 HloSchedule schedule(module.get()); in TEST_P()
711 HloSchedule schedule(module.get()); in TEST_P()
801 HloSchedule schedule(module.get()); in TEST_P()
871 HloSchedule schedule(module.get()); in TEST_P()
909 HloSchedule schedule(module.get()); in TEST_P()
946 HloSchedule schedule(module.get()); in TEST_P()
[all …]
|
| D | heap_simulator.cc | 43 const HloSchedule& schedule, in MinimumMemoryForModule()
86 const HloSchedule* schedule) { in MinimumMemoryForComputation()
99 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run()
125 HloSchedule schedule(computation.parent()); in Run()
141 const BufferValue::SizeFunction& size_fn, const HloSchedule* schedule, in Run()
337 const HloSchedule* schedule, in HeapSimulator()
|
| D | hlo_rematerialization.h | 118 HloSchedule* schedule,
|
| D | hlo_module.cc | 52 Status HloModule::set_schedule(HloSchedule schedule) { in set_schedule()
420 HloSchedule schedule, in CreateFromProto()
421 HloSchedule::CreateFromProto(module.get(), proto.schedule())); in CreateFromProto()
717 HloSchedule clone_schedule(module.get()); in Clone()
|
| D | hlo_ordering.cc | 486 SequentialHloOrdering::SequentialHloOrdering(const HloSchedule& schedule) in SequentialHloOrdering()
491 SequentialHloOrdering::SequentialHloOrdering(HloSchedule&& schedule) in SequentialHloOrdering()
|
| D | hlo_ordering_test.cc | 429 HloSchedule schedule(module.get()); in TEST_F()
484 HloSchedule schedule(module.get()); in TEST_F()
|
| D | hlo_live_range.cc | 23 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run()
|
| D | buffer_assignment.cc | 724 HloSchedule schedule(module_); in ComputeSummaryStats()
1291 HloSchedule schedule(&assignment->module()); in AssignBuffersForComputations()
1439 HloSchedule schedule(&assignment->module()); in AssignBuffersWithSequentialOrdering()
1634 HloSchedule schedule(module); in CreateAssignment()
|
| D | heap_simulator_test.cc | 90 HloSchedule schedule(module.get()); in TEST_F()
193 HloSchedule schedule(module.get()); in TEST_F()
308 HloSchedule schedule(module_.get()); in RunWholeModule()
|
| D | buffer_assignment_test.cc | 137 HloSchedule schedule(module); in RunBufferAssignmentWithInstructionSequence()
2039 HloSchedule schedule = in RunBufferAssignment()
2348 HloSchedule schedule, in TEST_F()
2611 HloSchedule schedule = in TEST_F()
|
| D | hlo_rematerialization.cc | 1831 HloComputation* computation, HloSchedule* schedule, in RematerializeComputation()
2072 HloSchedule saved_schedule = module->schedule(); in Run()
|
| D | hlo_dataflow_analysis_test.cc | 1394 HloSchedule schedule(module_.get()); in TEST_P()
1474 HloSchedule schedule(module_.get()); in TEST_P()
1616 HloSchedule schedule(module_.get()); in TEST_P()
|
| D | hlo_alias_analysis_test.cc | 1241 HloSchedule schedule(module_.get()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 601 TF_ASSIGN_OR_RETURN(HloSchedule schedule, in RunHloPassesAndBufferAssignement()
717 TF_ASSIGN_OR_RETURN(HloSchedule schedule, in RunBackend()
914 TF_ASSIGN_OR_RETURN(HloSchedule schedule, in CompileAheadOfTime()
|