| /external/pytorch/torch/csrc/jit/serialization/ |
| D | storage_context.h | 36 reinterpret_cast<void*>(storage.unsafeGetStorageImpl())); in operator()
43 return lhs.unsafeGetStorageImpl() == rhs.unsafeGetStorageImpl(); in operator()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | Resize.cpp | 179 resize_bytes_meta(storage.unsafeGetStorageImpl(), std::move(new_size_bytes)); in maybe_resize_storage_meta()
281 at::native::resize_bytes_cpu(storage.unsafeGetStorageImpl(), newsize.expect_int()); in resize_bytes_nocuda()
283 at::native::resize_bytes_meta(storage.unsafeGetStorageImpl(), newsize); in resize_bytes_nocuda()
311 storage.unsafeGetStorageImpl()->set_allocator( in resize_bytes_nocuda()
312 new_tensor.storage().unsafeGetStorageImpl()->allocator()); in resize_bytes_nocuda()
|
| D | TensorProperties.cpp | 120 if (self.storage().unsafeGetStorageImpl() == src.storage().unsafeGetStorageImpl() && in is_set_to()
|
| D | AutogradComposite.cpp | 95 c10::StorageImpl* self_storage = self.storage().unsafeGetStorageImpl(); in _lazy_clone()
|
| D | Resize.h | 62 resize_bytes_cpu(storage.unsafeGetStorageImpl(), new_size_bytes); in maybe_resize_storage_cpu()
|
| D | Copy.cpp | 326 auto self_storage = self.unsafeGetTensorImpl()->unsafe_storage().unsafeGetStorageImpl(); in copy()
|
| /external/pytorch/torch/csrc/ |
| D | Storage.cpp | 45 auto maybe_pyobj = _storage.unsafeGetStorageImpl()->pyobj_slot()->check_pyobj( in THPStorage_NewWithStorage()
81 storage.unsafeGetStorageImpl()->pyobj_slot()->init_pyobj( in THPStorage_NewWithStorage()
92 c10::StorageImpl* storage_impl = storage.unsafeGetStorageImpl(); in THPStorage_Wrap()
155 if (storage.unsafeGetStorageImpl()->pyobj_slot()->check_pyobj( in THPStorage_isPreservable()
172 c10::StorageImpl* storage_impl = storage.unsafeGetStorageImpl(); in THPStorage_tryPreserve()
501 at::StorageImpl* old_storage_impl = storage.unsafeGetStorageImpl(); in THPStorage_get()
699 return PyLong_FromVoidPtr(THPStorage_Unpack(self).unsafeGetStorageImpl()); in THPStorage_get_cdata()
740 THPStorage_Unpack(storage).unsafeGetStorageImpl(), "Got a null Storage"); in THPStorage_assertNotNull()
|
| D | StorageMethods.cpp | 160 at::native::resize_bytes_cuda(storage.unsafeGetStorageImpl(), size_bytes); in THPStorage_resize_()
446 storage.unsafeGetStorageImpl(), file, save_size, element_size); in THPStorage_writeFile()
456 storage.unsafeGetStorageImpl(), fd, save_size, element_size); in THPStorage_writeFile()
506 storage.unsafeGetStorageImpl()); in THPStorage_setFromFile()
527 storage.unsafeGetStorageImpl()); in THPStorage_setFromFile()
|
| D | StorageSharing.cpp | 129 storage.unsafeGetStorageImpl()->set_allocator(new_storage.allocator()); in THPStorage_shareFilename()
227 storage.unsafeGetStorageImpl()->set_allocator(new_storage.allocator()); in THPStorage_shareFd()
294 c10::StorageImpl* storage_impl = storage.unsafeGetStorageImpl(); in THPStorage_shareCuda()
579 c10::StorageImpl* storage = THPStorage_Unpack(self).unsafeGetStorageImpl(); in THPStorage_weakRef()
|
| /external/pytorch/aten/src/ATen/ |
| D | StorageUtils.cpp | 50 c10::StorageImpl* origStorageImpl = origStorage.unsafeGetStorageImpl(); in share_memory_()
51 c10::StorageImpl* newStorageImpl = newStorage.unsafeGetStorageImpl(); in share_memory_()
|
| D | FunctionalStorageImpl.cpp | 106 …original_storage_size_ = base.unsafeGetTensorImpl()->unsafe_storage().unsafeGetStorageImpl()->sym_…
|
| D | FunctionalTensorWrapper.cpp | 157 return static_cast<functionalization::FunctionalStorageImpl*>(storage_.unsafeGetStorageImpl()); in functional_storage_impl()
285 …auto curr_storage_size = value_.unsafeGetTensorImpl()->unsafe_storage().unsafeGetStorageImpl()->sy… in storage_resize_()
|
| /external/executorch/runtime/core/exec_aten/util/ |
| D | tensor_util_aten.cpp | 124 t_dst.unsafeGetTensorImpl()->unsafe_storage().unsafeGetStorageImpl(); in share_tensor_data()
148 .unsafeGetStorageImpl() in copy_tensor_data()
193 impl->unsafe_storage().unsafeGetStorageImpl()->reset(); in reset_data_ptr()
|
| /external/pytorch/torch/csrc/jit/runtime/static/ |
| D | memory_planner.cpp | 362 tensor->storage().unsafeGetStorageImpl()->reset(); in deallocateOutputTensors()
423 TORCH_DCHECK_EQ(storageImpl, tensor->storage().unsafeGetStorageImpl()); in allocateManagedTensors()
461 at::StorageImpl* tensorStorageImpl = storage.unsafeGetStorageImpl(); in deallocateManagedTensors()
498 tensor->storage().unsafeGetStorageImpl(), &storages_[group_idx]); in deallocateManagedTensors()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Resize.h | 25 resize_bytes_cuda(storage.unsafeGetStorageImpl(), new_size_bytes); in maybe_resize_storage_cuda()
|
| /external/pytorch/torch/csrc/inductor/ |
| D | resize_storage_bytes.cpp | 20 variable.storage().unsafeGetStorageImpl(), new_size.expect_int()); in resize_storage_bytes_()
|
| /external/pytorch/c10/core/ |
| D | RefcountedDeleter.cpp | 51 c10::StorageImpl* storage_impl = storage.unsafeGetStorageImpl(); in newStorageImplFromRefcountedDataPtr()
|
| D | Storage.h | 155 StorageImpl* unsafeGetStorageImpl() const noexcept { in unsafeGetStorageImpl() function
|
| /external/executorch/runtime/executor/ |
| D | tensor_parser_aten.cpp | 96 at::StorageImpl* storage = impl->unsafe_storage().unsafeGetStorageImpl(); in parseTensor()
|
| /external/pytorch/aten/src/ATen/native/mps/ |
| D | TensorFactory.cpp | 30 auto storage = self->storage().unsafeGetStorageImpl(); in maybe_resize_storage_mps()
|
| /external/pytorch/aten/src/ATen/test/ |
| D | ivalue_test.cpp | 114 ASSERT_EQ(emptyStorage.unsafeGetStorageImpl(), ivEmpty.toStorage().unsafeGetStorageImpl()); in TEST()
115 ASSERT_EQ(nonemptyStorage.unsafeGetStorageImpl(), ivNonempty.toStorage().unsafeGetStorageImpl()); in TEST()
|
| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | RNN.cpp | 344 …auto cx_ = hx_.storage().unsafeGetStorageImpl() == cx_tmp.storage().unsafeGetStorageImpl() ? at:… in mkldnn_rnn_layer_backward()
|
| /external/pytorch/torch/csrc/utils/ |
| D | tensor_numpy.cpp | 200 prepared_tensor.storage().unsafeGetStorageImpl()->set_resizable(false); in tensor_to_numpy()
|
| /external/pytorch/aten/src/ATen/core/ |
| D | ivalue.cpp | 316 …return rhs.isStorage() && lhs.toStorage().unsafeGetStorageImpl() == rhs.toStorage().unsafeGetStora… in equals()
795 return out << v.toStorage().unsafeGetStorageImpl(); in operator <<()
|
| /external/pytorch/torch/csrc/jit/mobile/compatibility/ |
| D | backport_manager.cpp | 105 tensor.storage().unsafeGetStorageImpl())); in write_archive_current()
|