Searched refs:AsDeviceMemoryBase (Results 1 – 8 of 8) sorted by relevance
20 se::DeviceMemoryBase MaybeOwningDeviceMemory::AsDeviceMemoryBase() { in AsDeviceMemoryBase() function in xla::MaybeOwningDeviceMemory22 return absl::get<OwningDeviceMemory>(mem_).AsDeviceMemoryBase(); in AsDeviceMemoryBase()37 mem_ = result.AsDeviceMemoryBase(); in Release()
53 se::DeviceMemoryBase AsDeviceMemoryBase();
103 se::DeviceMemoryBase AsDeviceMemoryBase() { in AsDeviceMemoryBase() function
55 se::DeviceMemory<Eigen::half>(owning_lhs_buffer.AsDeviceMemoryBase()); in CompareEqualFloatBuffers()57 se::DeviceMemory<Eigen::half>(owning_rhs_buffer.AsDeviceMemoryBase()); in CompareEqualFloatBuffers()
37 se::DeviceMemoryBase buffer_addr = allocated_buffer.AsDeviceMemoryBase(); in AllocateBytes()
112 owning_buffer.AsDeviceMemoryBase()); in Create()
56 se::DeviceMemoryBase buffer_addr = allocated_buffer.AsDeviceMemoryBase(); in AllocateBytes()
116 unowning_buffers[i] = owning_buffers[i].AsDeviceMemoryBase(); in CreateBufferTable()