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.cc83 const BufferValue::SizeFunction& size_function, in Run()
108 const BufferValue::SizeFunction& size_function, in ListScheduler()
405 const BufferValue::SizeFunction& size_function) { in SumLogicalBufferSizes()
417 const BufferValue::SizeFunction& size_function, in ScheduleComputationHelper()
439 const BufferValue::SizeFunction& size_function, in DFSMemoryScheduler()
546 const BufferValue::SizeFunction& size_function, in ListMemoryScheduler()
566 const BufferValue::SizeFunction& size_function, in PostOrderMemoryScheduler()
584 const BufferValue::SizeFunction& size_function, in DefaultMemoryScheduler()
642 const BufferValue::SizeFunction& size_function, int64* peak_memory) { in DefaultModuleScheduler()
697 HloModule* module, const BufferValue::SizeFunction& size_function, in ScheduleModule()
[all …]
Dhlo_rematerialization.h90 : size_function_(size_function), in size_function_() argument
Dheap_simulator.cc44 const LogicalBuffer::SizeFunction& size_function) { in MinimumMemoryForModule()
70 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation()
85 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation()
Dhlo_memory_scheduler_test.cc42 HloModule* module, LogicalBuffer::SizeFunction size_function) { in PeakMemoryUseOfEntryComputation()
Dhlo_rematerialization.cc783 const HloRematerialization::ShapeSizeFunction& size_function, in MemoryUsageTracker()