Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_memory_scheduler.cc83 const BufferValue::SizeFunction& size_function, in Run() argument
86 ListScheduler scheduler(computation, points_to_analysis, size_function, in Run()
108 const BufferValue::SizeFunction& size_function, in ListScheduler() argument
113 size_function_(size_function), in ListScheduler()
405 const BufferValue::SizeFunction& size_function) { in SumLogicalBufferSizes() argument
408 size += size_function(*buffer); in SumLogicalBufferSizes()
417 const BufferValue::SizeFunction& size_function, in ScheduleComputationHelper() argument
426 size_function, memory_by_computation, peak_memory); in ScheduleComputationHelper()
429 size_function, memory_by_computation, in ScheduleComputationHelper()
439 const BufferValue::SizeFunction& size_function, in DFSMemoryScheduler() argument
[all …]
Dhlo_memory_scheduler.h67 const LogicalBuffer::SizeFunction& size_function,
77 const LogicalBuffer::SizeFunction& size_function,
87 const LogicalBuffer::SizeFunction& size_function,
100 const LogicalBuffer::SizeFunction& size_function,
108 const LogicalBuffer::SizeFunction& size_function, int64* peak_memory);
115 HloModule* module, const LogicalBuffer::SizeFunction& size_function,
123 const LogicalBuffer::SizeFunction& size_function);
133 HloMemoryScheduler(const LogicalBuffer::SizeFunction& size_function,
Dhlo_rematerialization.h84 const ShapeSizeFunction& size_function, int64 memory_limit_bytes,
90 : size_function_(size_function), in size_function_() argument
Dheap_simulator.h117 const LogicalBuffer::SizeFunction& size_function);
124 const LogicalBuffer::SizeFunction& size_function,
131 const LogicalBuffer::SizeFunction& size_function,
Dheap_simulator.cc44 const LogicalBuffer::SizeFunction& size_function) { in MinimumMemoryForModule() argument
62 schedule, *alias_analysis, size_function)); in MinimumMemoryForModule()
70 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation() argument
77 sequence, alias_analysis, size_function, HeapSimulator::Options(), in MinimumMemoryForComputation()
85 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation() argument
91 sequence, alias_analysis, size_function, schedule, in MinimumMemoryForComputation()
Dhlo_memory_scheduler_test.cc42 HloModule* module, LogicalBuffer::SizeFunction size_function) { in PeakMemoryUseOfEntryComputation() argument
55 *computation, sequence, *alias_analysis, size_function) in PeakMemoryUseOfEntryComputation()
Dhlo_rematerialization.cc495 const HloRematerialization::ShapeSizeFunction& size_function,
783 const HloRematerialization::ShapeSizeFunction& size_function, in MemoryUsageTracker() argument
790 size_function_(size_function), in MemoryUsageTracker()