Searched refs:ScheduleModule (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_schedule_test.cc | 59 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
93 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
142 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
186 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
247 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
316 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
|
| D | hlo_memory_scheduler_test.cc | 127 ScheduleModule(module.get(), size_fn, ListMemoryScheduler)); in TEST_F()
178 ScheduleModule( in TEST_F()
229 ScheduleModule( in TEST_F()
289 ScheduleModule(module.get(), size_fn, ListMemoryScheduler)); in TEST_F()
|
| D | hlo_memory_scheduler.h | 81 StatusOr<HloSchedule> ScheduleModule(
|
| D | hlo_memory_scheduler.cc | 570 StatusOr<HloSchedule> ScheduleModule( in ScheduleModule() function
616 ScheduleModule(module, size_function_, algorithm_)); in Run()
|
| D | buffer_assignment_test.cc | 1891 ScheduleModule(module, ByteSizeOf).ConsumeValueOrDie(); in RunBufferAssignment()
2201 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
2430 ScheduleModule(module.get(), ByteSizeOf).ConsumeValueOrDie(); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 639 ScheduleModule(module.get(), BufferSizeBytesFunction(), in RunBackend()
824 ScheduleModule(module, BufferSizeBytesFunction())); in CompileAheadOfTime()
|