Searched defs:gpu_mem (Results 1 – 4 of 4) sorted by relevance
123 const GraphMemory::MemoryUsage& gpu_mem = memory.GetPeakMemoryUsage("/GPU:0"); in TEST_F() local149 const GraphMemory::MemoryUsage& gpu_mem = in TEST_F() local
135 void* gpu_mem = location->opaque(); in MemZero() local145 void* gpu_mem = location->opaque(); in Memset() local155 void* gpu_mem = location->opaque(); in Memset32() local
80 static hipDeviceptr_t AsROCmDevicePtr(const DeviceMemoryBase& gpu_mem) { in AsROCmDevicePtr()85 static hipDeviceptr_t AsROCmDevicePtr(DeviceMemoryBase* gpu_mem) { in AsROCmDevicePtr()
106 static CUdeviceptr AsCudaDevicePtr(const DeviceMemoryBase& gpu_mem) { in AsCudaDevicePtr()111 static CUdeviceptr AsCudaDevicePtr(DeviceMemoryBase* gpu_mem) { in AsCudaDevicePtr()