Searched defs:size_function (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_memory_scheduler.cc | 80 const LogicalBuffer::SizeFunction& size_function, in Run() 105 const LogicalBuffer::SizeFunction& size_function, in ListScheduler() 393 const LogicalBuffer::SizeFunction& size_function) { in SumLogicalBufferSizes() 404 const LogicalBuffer::SizeFunction& size_function, in ScheduleComputationHelper() 422 const LogicalBuffer::SizeFunction& size_function, in DFSMemoryScheduler() 493 const LogicalBuffer::SizeFunction& size_function, in ListMemoryScheduler() 503 const LogicalBuffer::SizeFunction& size_function, in PostOrderMemoryScheduler() 512 const LogicalBuffer::SizeFunction& size_function, in DefaultMemoryScheduler() 571 HloModule* module, const LogicalBuffer::SizeFunction& size_function, in ScheduleModule() 600 const LogicalBuffer::SizeFunction& size_function) { in ScheduleComputation() [all …]
|
D | hlo_rematerialization.h | 60 HloRematerialization(const ShapeSizeFunction& size_function, in HloRematerialization()
|
D | heap_simulator.cc | 35 const LogicalBuffer::SizeFunction& size_function) { in MinimumMemoryForModule() 60 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation()
|
D | hlo_rematerialization.cc | 416 const HloRematerialization::ShapeSizeFunction& size_function, in CreateBufferFromLogicalBuffer() 510 const HloRematerialization::ShapeSizeFunction& size_function, in MemoryUsageTracker()
|