Searched refs:SizeFunction (Results 1 – 11 of 11) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_memory_scheduler.h | 45 const LogicalBuffer::SizeFunction&, 53 const LogicalBuffer::SizeFunction&, 67 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, [all …]
|
D | hlo_memory_scheduler.cc | 83 const BufferValue::SizeFunction& size_function, in Run() 108 const BufferValue::SizeFunction& size_function, in ListScheduler() 384 const BufferValue::SizeFunction& size_function_; 405 const BufferValue::SizeFunction& size_function) { in SumLogicalBufferSizes() 417 const BufferValue::SizeFunction& size_function, in ScheduleComputationHelper() 439 const BufferValue::SizeFunction& size_function, in DFSMemoryScheduler() 520 const LogicalBuffer::SizeFunction& size_func, in ComputationSchedulerToModuleScheduler() 546 const BufferValue::SizeFunction& size_function, in ListMemoryScheduler() 566 const BufferValue::SizeFunction& size_function, in PostOrderMemoryScheduler() 584 const BufferValue::SizeFunction& size_function, in DefaultMemoryScheduler() [all …]
|
D | heap_simulator.h | 117 const LogicalBuffer::SizeFunction& size_function); 124 const LogicalBuffer::SizeFunction& size_function, 131 const LogicalBuffer::SizeFunction& size_function, 147 const BufferValue::SizeFunction& size_fn, 159 const BufferValue::SizeFunction& size_fn, 171 const BufferValue::SizeFunction& size_fn, const HloSchedule* schedule, 179 const BufferValue::SizeFunction& size_fn, 214 const BufferValue::SizeFunction size_fn_;
|
D | buffer_value.h | 95 using SizeFunction = std::function<int64(const BufferValue&)>; variable 152 LogicalBufferProto ToProto(const SizeFunction& size_fn) const;
|
D | buffer_assignment.h | 499 BufferValue::SizeFunction buffer_size, in BufferAssignment() 570 BufferValue::SizeFunction buffer_size_; 614 BufferValue::SizeFunction buffer_size, 635 BufferValue::SizeFunction buffer_size,
|
D | heap_simulator.cc | 44 const LogicalBuffer::SizeFunction& size_function) { in MinimumMemoryForModule() 70 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation() 85 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation() 100 const BufferValue::SizeFunction& size_fn, const Options& options) { in Run() 120 const BufferValue::SizeFunction& size_fn, const Options& options, in Run() 141 const BufferValue::SizeFunction& size_fn, const HloSchedule* schedule, in Run() 336 const BufferValue::SizeFunction& size_fn, const Options& options, in HeapSimulator()
|
D | buffer_value.cc | 54 LogicalBufferProto BufferValue::ToProto(const SizeFunction& size_fn) const { in ToProto()
|
D | hlo_memory_scheduler_test.cc | 42 HloModule* module, LogicalBuffer::SizeFunction size_function) { in PeakMemoryUseOfEntryComputation()
|
D | buffer_assignment.cc | 911 BufferValue::SizeFunction buffer_size, in Run() 1627 BufferValue::SizeFunction buffer_size, in CreateAssignment()
|
D | memory_space_assignment.h | 427 BufferValue::SizeFunction size_fn;
|
/external/llvm-project/compiler-rt/lib/asan/ |
D | asan_malloc_win.cpp | 272 using SizeFunction = size_t(WINAPI *)(HANDLE, DWORD, LPVOID); typedef 275 void *SharedReAlloc(ReAllocFunction reallocFunc, SizeFunction heapSizeFunc, in SharedReAlloc() 391 return SharedReAlloc(REAL(HeapReAlloc), (SizeFunction)REAL(HeapSize), in INTERCEPTOR_WINAPI()
|