Searched refs:ScheduleModule (Results 1 – 7 of 7) 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()
387 ScheduleModule(module.get(), in TEST_F()
|
| D | hlo_memory_scheduler_test.cc | 147 ScheduleModule(module.get(), size_fn, in TEST_F()
199 ScheduleModule(module.get(), size_fn, in TEST_F()
244 ScheduleModule( in TEST_F()
295 ScheduleModule( in TEST_F()
|
| D | hlo_memory_scheduler.h | 124 StatusOr<HloSchedule> ScheduleModule(
|
| D | hlo_memory_scheduler.cc | 723 StatusOr<HloSchedule> ScheduleModule( in ScheduleModule() function
768 ScheduleModule(module, size_function_, algorithm_, execution_threads)); in Run()
|
| D | buffer_assignment_test.cc | 2118 HloSchedule schedule = ScheduleModule(module, ByteSizeOf).value(); in RunBufferAssignment()
2427 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
2750 HloSchedule schedule = ScheduleModule(module.get(), ByteSizeOf).value(); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_hlo_schedule.cc | 156 return ScheduleModule( in ScheduleGpuModule()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 840 ScheduleModule(module, BufferSizeBytesFunction(), in AssignBuffers()
1168 ScheduleModule(module.get(), BufferSizeBytesFunction(), in CompileLegacyCpuExecutable()
1421 ScheduleModule(module, BufferSizeBytesFunction())); in CompileAheadOfTime()
|