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()
440 const BufferValue::SizeFunction& size_function, in DFSMemoryScheduler()
562 const BufferValue::SizeFunction& size_function, in ListMemoryScheduler()
585 const BufferValue::SizeFunction& size_function, in PostOrderMemoryScheduler()
606 const BufferValue::SizeFunction& size_function, in DefaultMemoryScheduler()
666 const BufferValue::SizeFunction& size_function, in DefaultModuleScheduler()
724 const HloModule* module, const BufferValue::SizeFunction& size_function, in ScheduleModule()
[all …]
Dhlo_rematerialization.h91 : size_function_(size_function), in size_function_() argument
Dheap_simulator.cc50 const LogicalBuffer::SizeFunction& size_function) { in MinimumMemoryForModule()
75 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation()
89 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation()
Dhlo_memory_scheduler_test.cc42 HloModule* module, LogicalBuffer::SizeFunction size_function) { in PeakMemoryUseOfEntryComputation()
Dhlo_rematerialization.cc794 const HloRematerialization::ShapeSizeFunction& size_function, in MemoryUsageTracker()