Home
last modified time | relevance | path

Searched refs:SizeFunction (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_memory_scheduler.h40 const LogicalBuffer::SizeFunction&,
48 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,
107 LogicalBuffer::SizeFunction size_function_;
Dhlo_memory_scheduler.cc80 const LogicalBuffer::SizeFunction& size_function, in Run()
105 const LogicalBuffer::SizeFunction& size_function, in ListScheduler()
372 const LogicalBuffer::SizeFunction& size_function_;
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()
[all …]
Dheap_simulator.h95 const LogicalBuffer::SizeFunction& size_function);
102 const LogicalBuffer::SizeFunction& size_function,
119 const BufferValue::SizeFunction& size_fn,
131 const BufferValue::SizeFunction& size_fn,
141 const BufferValue::SizeFunction& size_fn,
168 const BufferValue::SizeFunction size_fn_;
Dbuffer_assignment.h475 LogicalBuffer::SizeFunction buffer_size, in BufferAssignment()
525 LogicalBuffer::SizeFunction buffer_size_;
550 LogicalBuffer::SizeFunction buffer_size,
569 LogicalBuffer::SizeFunction buffer_size,
619 const LogicalBuffer::SizeFunction& buffer_size,
642 const LogicalBuffer::SizeFunction& buffer_size);
Dbuffer_value.h96 using SizeFunction = std::function<int64(const BufferValue&)>; variable
150 LogicalBufferProto ToProto(const SizeFunction& size_fn) const;
Dbuffer_value.cc54 LogicalBufferProto BufferValue::ToProto(const SizeFunction& size_fn) const { in ToProto()
Dheap_simulator.cc35 const LogicalBuffer::SizeFunction& size_function) { in MinimumMemoryForModule()
60 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation()
77 const BufferValue::SizeFunction& size_fn, const Options& options) { in Run()
92 const BufferValue::SizeFunction& size_fn, const Options& options, in Run()
351 const BufferValue::SizeFunction& size_fn, const Options& options, in HeapSimulator()
Dbuffer_assignment.cc737 LogicalBuffer::SizeFunction buffer_size, in Run()
827 const LogicalBuffer::SizeFunction& buffer_size = assignment->buffer_size_; in MaybeAssignBuffer()
1420 const LogicalBuffer::SizeFunction& buffer_size) { in MergeColocatedBufferSets()
1532 const LogicalBuffer::SizeFunction& buffer_size, in BuildColocatedBufferSets()
1756 LogicalBuffer::SizeFunction buffer_size, in CreateAssignment()