Searched refs:dst_device_buffer (Results 1 – 1 of 1) sorted by relevance
452 std::shared_ptr<TrackedDeviceBuffer> dst_device_buffer = in AllocateDestinationBuffer() local457 on_device_shape, std::move(dst_device_buffer), client, device); in AllocateDestinationBuffer()1307 ScopedHold dst_device_buffer(py_buffer->GetBufferWithUsageHold()); in CopyToDeviceHelper() local1308 CHECK(dst_device_buffer.ok()); in CopyToDeviceHelper()1309 ShapedBuffer dst_buffer = dst_device_buffer->AsShapedBuffer(on_device_shape_); in CopyToDeviceHelper()1328 dst_device_buffer->definition_events()[0]; in CopyToDeviceHelper()1330 transfer_local_device, std::move(dst_device_buffer), event, in CopyToDeviceHelper()