Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/c10/core/impl/
H A DPyInterpreter.h202 virtual void trace_gpu_memory_deallocation(
H A DPyInterpreter.cpp116 void trace_gpu_memory_deallocation(c10::DeviceType device_type, uintptr_t ptr) in trace_gpu_memory_deallocation() function
/aosp_15_r20/external/pytorch/c10/xpu/
H A DXPUCachingAllocator.cpp579 (*interp)->trace_gpu_memory_deallocation( in free()
/aosp_15_r20/external/pytorch/torch/csrc/
H A DPyInterpreter.cpp123 void trace_gpu_memory_deallocation(at::DeviceType device_type, uintptr_t ptr) in trace_gpu_memory_deallocation() function
/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDACachingAllocator.cpp3142 (*interp)->trace_gpu_memory_deallocation( in uncached_delete()
3244 (*interp)->trace_gpu_memory_deallocation( in free()