Searched defs:GetDeviceAddress (Results 1 – 2 of 2) sorted by relevance
58 se::DeviceMemoryBase BufferAllocations::GetDeviceAddress( in GetDeviceAddress() function in xla::gpu::BufferAllocations72 se::DeviceMemoryBase BufferAllocations::GetDeviceAddress( in GetDeviceAddress() function in xla::gpu::BufferAllocations
236 static se::DeviceMemoryBase GetDeviceAddress(MemrefArg& memref) { in GetDeviceAddress() function242 static se::DeviceMemoryBase GetDeviceAddress(runtime::FlatMemrefView& memref) { in GetDeviceAddress() function