Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_memory_scheduler.h45 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 …]
Dhlo_memory_scheduler.cc83 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 …]
Dheap_simulator.h117 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_;
Dbuffer_value.h95 using SizeFunction = std::function<int64(const BufferValue&)>; variable
152 LogicalBufferProto ToProto(const SizeFunction& size_fn) const;
Dbuffer_assignment.h499 BufferValue::SizeFunction buffer_size, in BufferAssignment()
570 BufferValue::SizeFunction buffer_size_;
614 BufferValue::SizeFunction buffer_size,
635 BufferValue::SizeFunction buffer_size,
Dheap_simulator.cc44 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()
Dbuffer_value.cc54 LogicalBufferProto BufferValue::ToProto(const SizeFunction& size_fn) const { in ToProto()
Dhlo_memory_scheduler_test.cc42 HloModule* module, LogicalBuffer::SizeFunction size_function) { in PeakMemoryUseOfEntryComputation()
Dbuffer_assignment.cc911 BufferValue::SizeFunction buffer_size, in Run()
1627 BufferValue::SizeFunction buffer_size, in CreateAssignment()
Dmemory_space_assignment.h427 BufferValue::SizeFunction size_fn;
/external/llvm-project/compiler-rt/lib/asan/
Dasan_malloc_win.cpp272 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()