Home
last modified time | relevance | path

Searched refs:trace_gpu_memory_allocation (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/c10/core/impl/
H A DPyInterpreter.h199 virtual void trace_gpu_memory_allocation(
H A DPyInterpreter.cpp114 void trace_gpu_memory_allocation(c10::DeviceType device_type, uintptr_t ptr) in trace_gpu_memory_allocation() function
/aosp_15_r20/external/pytorch/c10/xpu/
H A DXPUCachingAllocator.cpp565 (*interp)->trace_gpu_memory_allocation( in malloc()
/aosp_15_r20/external/pytorch/torch/csrc/
H A DPyInterpreter.cpp119 void trace_gpu_memory_allocation(at::DeviceType device_type, uintptr_t ptr) in trace_gpu_memory_allocation() function
/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDACachingAllocator.cpp3229 (*interp)->trace_gpu_memory_allocation( in malloc()
3468 (*interp)->trace_gpu_memory_allocation( in allocate()