| 142 | } |
| 143 | |
| 144 | Maybe<bool> EagerLocalTensorImpl::is_pinned() const { |
| 145 | if (this->device() == JUST(Device::New("meta"))) { return false; } |
| 146 | if (!eager_blob_object_) { return false; } |
| 147 | return IsStreamAllocatorPinned::Visit(JUST(eager_blob_object_->producer_stream())->stream_type()); |
| 148 | } |
| 149 | |
| 150 | Maybe<void> EagerLocalTensorImpl::set_eager_blob_object( |
| 151 | std::shared_ptr<vm::EagerBlobObject> eager_blob_object) { |
no test coverage detected