| /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.h | 81 class HloSchedule {
83 explicit HloSchedule(const HloModule* module) : module_(module) {} in HloSchedule() function
86 static StatusOr<HloSchedule> CreateFromProto(const HloModule* module,
154 std::ostream& operator<<(std::ostream& out, const HloSchedule& schedule);
|
| D | hlo_schedule_test.cc | 58 HloSchedule schedule, in TEST_F()
92 HloSchedule schedule, in TEST_F()
141 HloSchedule schedule, in TEST_F()
185 HloSchedule schedule, in TEST_F()
246 HloSchedule schedule, in TEST_F()
315 HloSchedule schedule, in TEST_F()
|
| D | hlo_module.h | 262 Status set_schedule(HloSchedule schedule);
271 const HloSchedule& schedule() const { return *schedule_; } in schedule()
272 HloSchedule& schedule() { return *schedule_; } in schedule()
317 absl::optional<HloSchedule> schedule_;
|
| D | hlo_ordering.h | 184 SequentialHloOrdering(const HloSchedule& schedule);
185 SequentialHloOrdering(HloSchedule&& schedule);
200 const HloSchedule schedule_;
|
| D | heap_simulator.h | 94 const HloSchedule& schedule,
117 const HloSchedule& schedule,
142 const Options& options, const HloSchedule* schedule = nullptr,
174 const HloSchedule* schedule_;
|
| D | hlo_memory_scheduler_test.cc | 126 HloSchedule schedule, in TEST_F()
177 TF_ASSERT_OK_AND_ASSIGN(HloSchedule schedule, in TEST_F()
228 TF_ASSERT_OK_AND_ASSIGN(HloSchedule schedule, in TEST_F()
288 HloSchedule schedule, in TEST_F()
|
| D | hlo_memory_scheduler.cc | 570 StatusOr<HloSchedule> ScheduleModule( in ScheduleModule()
573 HloSchedule schedule(module); in ScheduleModule()
615 TF_ASSIGN_OR_RETURN(HloSchedule schedule, in Run()
622 HloSchedule schedule(module); in Run()
|
| D | hlo_module.cc | 44 Status HloModule::set_schedule(HloSchedule schedule) { in set_schedule()
360 HloSchedule schedule, in CreateFromProto()
361 HloSchedule::CreateFromProto(module.get(), proto.schedule())); in CreateFromProto()
574 HloSchedule clone_schedule(module.get()); in Clone()
|
| D | hlo_rematerialization.h | 83 HloSchedule* schedule,
|
| D | hlo_memory_scheduler.h | 81 StatusOr<HloSchedule> ScheduleModule(
|
| D | hlo_ordering.cc | 349 SequentialHloOrdering::SequentialHloOrdering(const HloSchedule& schedule) in SequentialHloOrdering()
354 SequentialHloOrdering::SequentialHloOrdering(HloSchedule&& schedule) in SequentialHloOrdering()
|
| D | hlo_ordering_test.cc | 423 HloSchedule schedule(module.get()); in TEST_F()
478 HloSchedule schedule(module.get()); in TEST_F()
|
| D | buffer_liveness_test.cc | 170 HloSchedule schedule(module.get()); in TEST_F()
295 HloSchedule schedule(module.get()); in TEST_F()
342 HloSchedule schedule(module.get()); in TEST_F()
|
| D | heap_simulator.cc | 34 const HloSchedule& schedule, in MinimumMemoryForModule()
75 const HloSchedule& schedule, in Run()
352 const HloSchedule* schedule, in HeapSimulator()
|
| D | heap_simulator_test.cc | 89 HloSchedule schedule(module.get()); in TEST_F()
192 HloSchedule schedule(module.get()); in TEST_F()
295 HloSchedule schedule(module_.get()); in RunWholeModule()
|
| D | buffer_assignment_test.cc | 143 HloSchedule schedule(module); in RunBufferAssignmentWithInstructionSequence()
1890 HloSchedule schedule = in RunBufferAssignment()
2200 HloSchedule schedule, in TEST_F()
2429 HloSchedule schedule = in TEST_F()
|
| D | buffer_assignment.cc | 632 HloSchedule schedule(module_); in ComputeSummaryStats()
1173 HloSchedule schedule(&assignment->module()); in AssignBuffersWithSequentialOrdering()
|
| D | hlo_rematerialization.cc | 972 HloComputation* computation, HloSchedule* schedule, in RematerializeComputation()
|
| D | hlo_dataflow_analysis_test.cc | 1286 HloSchedule schedule(module_.get()); in TEST_P()
1366 HloSchedule schedule(module_.get()); in TEST_P()
1508 HloSchedule schedule(module_.get()); in TEST_P()
|
| D | hlo_alias_analysis_test.cc | 1089 HloSchedule schedule(module_.get()); in TEST_F()
|
| D | hlo_parser.cc | 54 HloSchedule ScheduleFromInstructionOrder(HloModule* module) { in ScheduleFromInstructionOrder()
55 HloSchedule schedule(module); in ScheduleFromInstructionOrder()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 638 TF_ASSIGN_OR_RETURN(HloSchedule schedule, in RunBackend()
823 TF_ASSIGN_OR_RETURN(HloSchedule schedule, in CompileAheadOfTime()
|