Home
last modified time | relevance | path

Searched defs:size_function (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_memory_scheduler.cc82 const BufferValue::SizeFunction& size_function, in Run()
107 const BufferValue::SizeFunction& size_function, in ListScheduler()
393 const BufferValue::SizeFunction& size_function) { in SumLogicalBufferSizes()
405 const BufferValue::SizeFunction& size_function, in ScheduleComputationHelper()
427 const BufferValue::SizeFunction& size_function, in DFSMemoryScheduler()
534 const BufferValue::SizeFunction& size_function, in ListMemoryScheduler()
554 const BufferValue::SizeFunction& size_function, in PostOrderMemoryScheduler()
572 const BufferValue::SizeFunction& size_function, in DefaultMemoryScheduler()
630 const BufferValue::SizeFunction& size_function, int64* peak_memory) { in DefaultModuleScheduler()
685 HloModule* module, const BufferValue::SizeFunction& size_function, in ScheduleModule()
[all …]
Dhlo_rematerialization.h88 : size_function_(size_function), in size_function_() argument
Dheap_simulator.cc43 const LogicalBuffer::SizeFunction& size_function) { in MinimumMemoryForModule()
68 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation()
82 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation()
Dhlo_memory_scheduler_test.cc42 HloModule* module, LogicalBuffer::SizeFunction size_function) { in PeakMemoryUseOfEntryComputation()
Dhlo_rematerialization.cc622 const HloRematerialization::ShapeSizeFunction& size_function, in MemoryUsageTracker()