Searched defs:size_function (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_memory_scheduler.cc | 83 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 …]
|
D | hlo_rematerialization.h | 90 : size_function_(size_function), in size_function_() argument
|
D | heap_simulator.cc | 44 const LogicalBuffer::SizeFunction& size_function) { in MinimumMemoryForModule() 70 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation() 85 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation()
|
D | hlo_memory_scheduler_test.cc | 42 HloModule* module, LogicalBuffer::SizeFunction size_function) { in PeakMemoryUseOfEntryComputation()
|
D | hlo_rematerialization.cc | 783 const HloRematerialization::ShapeSizeFunction& size_function, in MemoryUsageTracker()
|