Searched defs:gpu_mem (Results 1 – 4 of 4) sorted by relevance
102 void *gpu_mem = location->opaque(); in MemZero() local112 void *gpu_mem = location->opaque(); in Memset() local122 void *gpu_mem = location->opaque(); in Memset32() local
123 const GraphMemory::MemoryUsage& gpu_mem = memory.GetPeakMemoryUsage("/GPU:0"); in TEST_F() local149 const GraphMemory::MemoryUsage& gpu_mem = in TEST_F() local
81 static hipDeviceptr_t AsROCmDevicePtr(const DeviceMemoryBase& gpu_mem) { in AsROCmDevicePtr()86 static hipDeviceptr_t AsROCmDevicePtr(DeviceMemoryBase* gpu_mem) { in AsROCmDevicePtr()
104 static CUdeviceptr AsCudaDevicePtr(const DeviceMemoryBase &gpu_mem) { in AsCudaDevicePtr()109 static CUdeviceptr AsCudaDevicePtr(DeviceMemoryBase *gpu_mem) { in AsCudaDevicePtr()