Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_memory_scheduler.cc83 const BufferValue::SizeFunction& size_function, in Run()
108 const BufferValue::SizeFunction& size_function, in ListScheduler()
405 const BufferValue::SizeFunction& size_function) { in SumLogicalBufferSizes()
417 const BufferValue::SizeFunction& size_function, in ScheduleComputationHelper()
440 const BufferValue::SizeFunction& size_function, in DFSMemoryScheduler()
562 const BufferValue::SizeFunction& size_function, in ListMemoryScheduler()
585 const BufferValue::SizeFunction& size_function, in PostOrderMemoryScheduler()
606 const BufferValue::SizeFunction& size_function, in DefaultMemoryScheduler()
666 const BufferValue::SizeFunction& size_function, in DefaultModuleScheduler()
724 const HloModule* module, const BufferValue::SizeFunction& size_function, in ScheduleModule()
[all …]
H A Dheap_simulator.cc50 const LogicalBuffer::SizeFunction& size_function) { in MinimumMemoryForModule()
75 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation()
89 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation()
H A Dhlo_rematerialization.h91 : size_function_(size_function), in size_function_() argument
H A Dhlo_memory_scheduler_test.cc42 HloModule* module, LogicalBuffer::SizeFunction size_function) { in PeakMemoryUseOfEntryComputation()
H A Dhlo_rematerialization.cc794 const HloRematerialization::ShapeSizeFunction& size_function, in MemoryUsageTracker()
/aosp_15_r20/system/logging/logd/
H A DCommandListener.cpp99 static int LogSizeCommand(SocketClient* cli, int argc, char** argv, F&& size_function) { in LogSizeCommand()
/aosp_15_r20/external/mesa3d/src/gallium/auxiliary/gallivm/
H A Dlp_bld_jit_types.h221 void *size_function; member