Searched refs:tracked_device_buffer_ (Results 1 – 1 of 1) sorted by relevance
617 : tracked_device_buffer_(std::move(tracked_device_buffer)) { in TrackedDeviceBufferExternalReference()618 data_ptr_ = tracked_device_buffer_->device_memory()[0].opaque(); in TrackedDeviceBufferExternalReference()624 std::shared_ptr<TrackedDeviceBuffer> tracked_device_buffer_; member in xla::TrackedDeviceBufferExternalReference