Home
last modified time | relevance | path

Searched refs:MemorySchedulerPostprocessor (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_memory_scheduler.h38 using MemorySchedulerPostprocessor = variable
53 const MemorySchedulerPostprocessor&,
68 const MemorySchedulerAlgorithm&, const MemorySchedulerPostprocessor& = {});
78 const MemorySchedulerPostprocessor& postprocessor, int64_t* peak_memory);
88 const MemorySchedulerPostprocessor& postprocessor, int64_t* peak_memory);
98 const MemorySchedulerPostprocessor& postprocessor, int64_t* peak_memory);
111 const MemorySchedulerPostprocessor& postprocessor, int64_t* peak_memory);
135 const MemorySchedulerPostprocessor& postprocessor);
H A Dhlo_memory_scheduler.cc421 const MemorySchedulerPostprocessor& postprocessor, int64_t* peak_memory) { in ScheduleComputationHelper()
443 const MemorySchedulerPostprocessor& postprocessor, int64_t* peak_memory) { in DFSMemoryScheduler()
528 const MemorySchedulerPostprocessor& postprocessor) { in ComputationSchedulerToModuleScheduler()
565 const MemorySchedulerPostprocessor& postprocessor, int64_t* peak_memory) { in ListMemoryScheduler()
588 const MemorySchedulerPostprocessor& postprocessor, int64_t* peak_memory) { in PostOrderMemoryScheduler()
609 const MemorySchedulerPostprocessor& postprocessor, int64_t* peak_memory) { in DefaultMemoryScheduler()
745 const MemorySchedulerPostprocessor& postprocessor) { in ScheduleComputation()