Searched refs:trace_gpu_memory_allocation (Results 1 – 5 of 5) sorted by relevance
199 virtual void trace_gpu_memory_allocation(
114 void trace_gpu_memory_allocation(c10::DeviceType device_type, uintptr_t ptr) in trace_gpu_memory_allocation() function
565 (*interp)->trace_gpu_memory_allocation( in malloc()
119 void trace_gpu_memory_allocation(at::DeviceType device_type, uintptr_t ptr) in trace_gpu_memory_allocation() function
3229 (*interp)->trace_gpu_memory_allocation( in malloc()3468 (*interp)->trace_gpu_memory_allocation( in allocate()