Searched refs:MaybeOwningDeviceMemory (Results 1 – 5 of 5) sorted by relevance
29 class MaybeOwningDeviceMemory {31 MaybeOwningDeviceMemory() = default;32 explicit MaybeOwningDeviceMemory(OwningDeviceMemory owned) in MaybeOwningDeviceMemory() function34 explicit MaybeOwningDeviceMemory(se::DeviceMemoryBase unowned) in MaybeOwningDeviceMemory() function36 MaybeOwningDeviceMemory(MaybeOwningDeviceMemory&&) = default;37 ~MaybeOwningDeviceMemory() = default;39 MaybeOwningDeviceMemory& operator=(se::DeviceMemoryBase unowned) {44 MaybeOwningDeviceMemory& operator=(OwningDeviceMemory owned) {49 MaybeOwningDeviceMemory& operator=(MaybeOwningDeviceMemory&&) = default;
20 se::DeviceMemoryBase MaybeOwningDeviceMemory::AsDeviceMemoryBase() { in AsDeviceMemoryBase()28 bool MaybeOwningDeviceMemory::HasOwnership() const { in HasOwnership()32 absl::optional<OwningDeviceMemory> MaybeOwningDeviceMemory::Release() { in Release()
104 std::vector<ShapeTree<xla::MaybeOwningDeviceMemory>> arguments, in ExecuteOnStream()112 std::vector<ShapeTree<xla::MaybeOwningDeviceMemory>> arguments) { in ExecuteAsyncOnStream()
189 xla::ShapeTree<xla::MaybeOwningDeviceMemory> ToDeviceMemoryTree(
540 xla::ShapeTree<xla::MaybeOwningDeviceMemory>543 xla::ShapeTree<xla::MaybeOwningDeviceMemory> shaped_tree(on_device_shape()); in ToDeviceMemoryTree()