Searched defs:gpu_mem (Results 1 – 5 of 5) 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
104 static CUdeviceptr AsCudaDevicePtr(const DeviceMemoryBase& gpu_mem) { in AsCudaDevicePtr()109 static CUdeviceptr AsCudaDevicePtr(DeviceMemoryBase* gpu_mem) { in AsCudaDevicePtr()
80 static hipDeviceptr_t AsROCmDevicePtr(const DeviceMemoryBase& gpu_mem) { in AsROCmDevicePtr()85 static hipDeviceptr_t AsROCmDevicePtr(DeviceMemoryBase* gpu_mem) { in AsROCmDevicePtr()
390 for (auto &gpu_mem : gpu_proc.memories()) { in init_compute_and_memory() local