Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_memory_scheduler.cc80 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 …]
Dhlo_rematerialization.h60 HloRematerialization(const ShapeSizeFunction& size_function, in HloRematerialization()
Dheap_simulator.cc35 const LogicalBuffer::SizeFunction& size_function) { in MinimumMemoryForModule()
60 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation()
Dhlo_rematerialization.cc416 const HloRematerialization::ShapeSizeFunction& size_function, in CreateBufferFromLogicalBuffer()
510 const HloRematerialization::ShapeSizeFunction& size_function, in MemoryUsageTracker()