Searched refs:GetMutableDeviceAddress (Results 1 – 3 of 3) sorted by relevance
62 se::DeviceMemoryBase& GetMutableDeviceAddress(
67 se::DeviceMemoryBase& BufferAllocations::GetMutableDeviceAddress( in GetMutableDeviceAddress() function in xla::gpu::BufferAllocations
525 buffer_allocations.GetMutableDeviceAddress( in ExecuteAsyncOnStreamImpl()