Searched defs:trace_gpu_memory_allocation (Results 1 – 2 of 2) sorted by relevance
114 void trace_gpu_memory_allocation(c10::DeviceType device_type, uintptr_t ptr) in trace_gpu_memory_allocation() function
119 void trace_gpu_memory_allocation(at::DeviceType device_type, uintptr_t ptr) in trace_gpu_memory_allocation() function