Home
last modified time | relevance | path

Searched refs:size_function (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_memory_scheduler.cc80 const LogicalBuffer::SizeFunction& size_function, in Run() argument
83 ListScheduler scheduler(computation, points_to_analysis, size_function, in Run()
105 const LogicalBuffer::SizeFunction& size_function, in ListScheduler() argument
110 size_function_(size_function), in ListScheduler()
393 const LogicalBuffer::SizeFunction& size_function) { in SumLogicalBufferSizes() argument
396 size += size_function(*buffer); in SumLogicalBufferSizes()
404 const LogicalBuffer::SizeFunction& size_function, in ScheduleComputationHelper() argument
410 return algorithm(computation, points_to_analysis, size_function, in ScheduleComputationHelper()
413 return DefaultMemoryScheduler(computation, points_to_analysis, size_function, in ScheduleComputationHelper()
422 const LogicalBuffer::SizeFunction& size_function, in DFSMemoryScheduler() argument
[all …]
Dhlo_memory_scheduler.h48 const LogicalBuffer::SizeFunction& size_function,
56 const LogicalBuffer::SizeFunction& size_function,
64 const LogicalBuffer::SizeFunction& size_function,
74 const LogicalBuffer::SizeFunction& size_function,
82 HloModule* module, const LogicalBuffer::SizeFunction& size_function,
89 const LogicalBuffer::SizeFunction& size_function);
99 HloMemoryScheduler(const LogicalBuffer::SizeFunction& size_function,
Dhlo_rematerialization.h60 HloRematerialization(const ShapeSizeFunction& size_function, in HloRematerialization() argument
62 : size_function_(size_function), in HloRematerialization()
Dheap_simulator.h95 const LogicalBuffer::SizeFunction& size_function);
102 const LogicalBuffer::SizeFunction& size_function,
Dhlo_rematerialization.cc329 const HloRematerialization::ShapeSizeFunction& size_function,
416 const HloRematerialization::ShapeSizeFunction& size_function, in CreateBufferFromLogicalBuffer() argument
422 size_function(logical_buffer->shape()), std::move(users), in CreateBufferFromLogicalBuffer()
510 const HloRematerialization::ShapeSizeFunction& size_function, in MemoryUsageTracker() argument
555 logical_buffer, points_to_analysis, size_function, in MemoryUsageTracker()
Dheap_simulator.cc35 const LogicalBuffer::SizeFunction& size_function) { in MinimumMemoryForModule() argument
52 schedule, *points_to_analysis, size_function)); in MinimumMemoryForModule()
60 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation() argument
67 size_function, HeapSimulator::Options(), in MinimumMemoryForComputation()