Searched defs:gpu_mem (Results 1 – 5 of 5) sorted by relevance
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
123 const GraphMemory::MemoryUsage& gpu_mem = memory.GetPeakMemoryUsage("/GPU:0"); in TEST_F() local149 const GraphMemory::MemoryUsage& gpu_mem = in TEST_F() local
404 auto gpu_mem = map.readValue(key); in ReadPerProcessGpuMem() local431 uint64_t gpu_mem; in ReadProcessGpuUsageKb() 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()