Searched refs:ScheduleModule (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_schedule_test.cc | 58 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
92 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
141 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
185 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
246 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
315 ScheduleModule(module.get(), [](const BufferValue& buffer) { 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()
245 ScheduleModule( in TEST_F()
296 ScheduleModule( in TEST_F()
|
| D | hlo_memory_scheduler.h | 114 StatusOr<HloSchedule> ScheduleModule(
|
| D | hlo_memory_scheduler.cc | 696 StatusOr<HloSchedule> ScheduleModule( in ScheduleModule() function
735 ScheduleModule(module, size_function_, algorithm_)); in Run()
|
| D | buffer_assignment_test.cc | 2040 ScheduleModule(module, ByteSizeOf).ConsumeValueOrDie(); in RunBufferAssignment()
2349 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
2612 ScheduleModule(module.get(), ByteSizeOf).ConsumeValueOrDie(); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 602 ScheduleModule(module.get(), BufferSizeBytesFunction(), in RunHloPassesAndBufferAssignement()
718 ScheduleModule(module.get(), BufferSizeBytesFunction(), in RunBackend()
915 ScheduleModule(module, BufferSizeBytesFunction())); in CompileAheadOfTime()
|