| /external/executorch/kernels/test/ |
| D | op_as_strided_copy_test.cpp | 34 optional<int64_t> storage_offset, in op_as_strided_copy_out()
51 optional<int64_t> storage_offset; in test_detach_copy_out() local
81 optional<int64_t> storage_offset; in test_as_strided_copy_out_invalid_parameters() local
152 optional<int64_t> storage_offset = 2; in test_detach_copy_out() local
177 optional<int64_t> storage_offset = 2; in test_detach_copy_out() local
217 optional<int64_t> storage_offset; in TEST_F() local
279 optional<int64_t> storage_offset; in TEST_F() local
320 optional<int64_t> storage_offset; in TEST_F() local
364 optional<int64_t> storage_offset; in TEST_F() local
|
| /external/pytorch/aten/src/ATen/native/ |
| D | Resize.h | 85 T storage_offset, in checkInBoundsForStorage()
114 inline void checkSetStorage(Tensor& result, Storage storage, T storage_offset, in checkSetStorage()
156 T storage_offset) { in setStrided()
|
| D | Fill.cpp | 112 int64_t storage_offset = self.storage_offset(); in fill_diagonal_() local
|
| D | TensorShape.cpp | 379 Tensor& set_storage_cpu_(Tensor& result, Storage storage, int64_t storage_offset, IntArrayRef size,… in set_storage_cpu_()
391 Tensor& set_storage_meta__symint(Tensor& result, Storage storage, c10::SymInt storage_offset, c10::… in set_storage_meta__symint()
441 Tensor& set__symint(Tensor& result, const Tensor& storage, c10::SymInt storage_offset, c10::SymIntA… in set__symint()
1064 int64_t storage_offset = self.storage_offset(); in diagonal() local
1177 auto storage_offset = storage_offset_.value_or(self.storage_offset()); in as_strided_tensorimpl() local
1189 T&& storage_offset) { in setStridedUnchecked()
1209 auto storage_offset = storage_offset_.value_or(self.storage_offset()); in as_strided_qtensorimpl() local
1227 auto storage_offset = storage_offset_.value_or(self.storage_offset()); in as_strided_qtensorimpl() local
1239 auto storage_offset = storage_offset_.value_or(self.sym_storage_offset()); in as_strided__symint() local
1849 auto storage_offset = self.storage_offset() + local_index * strides[dim]; in select_symint() local
[all …]
|
| D | Resize.cpp | 207 const auto storage_offset = self->generic_storage_offset<T>(); in _resize_impl_() local
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Resize.h | 37 const auto storage_offset = self->storage_offset(); in resize_impl_cuda_() local
|
| D | TensorShapeCUDA.cpp | 31 Tensor& set_storage_cuda_(Tensor& result, Storage storage, int64_t storage_offset, IntArrayRef size… in set_storage_cuda_()
|
| /external/pytorch/torch/ |
| D | _utils.py | 172 def _rebuild_tensor(storage, storage_offset, size, stride): argument
195 storage_offset, argument
216 storage_offset, argument
355 storage_offset, argument
377 storage_offset, argument
|
| /external/pytorch/torch/csrc/inductor/aoti_runtime/ |
| D | utils.h | 108 int64_t storage_offset() { in storage_offset() function
109 int64_t storage_offset = 0; in storage_offset() local
|
| D | arrayref_tensor.h | 283 int64_t storage_offset) { in reinterpret_tensor_wrapper()
296 int64_t storage_offset) { in reinterpret_tensor_wrapper()
|
| /external/executorch/kernels/portable/cpu/ |
| D | op_as_strided_copy.cpp | 24 optional<int64_t> storage_offset, in as_strided_copy_out()
|
| D | op_diagonal_copy.cpp | 30 int64_t storage_offset = 0; in diagonal_copy_impl() local
|
| /external/pytorch/benchmarks/operator_benchmark/pt/ |
| D | as_strided_test.py | 37 def init(self, M, N, size, stride, storage_offset, device): argument
|
| /external/pytorch/aten/src/ATen/ |
| D | EmptyTensor.cpp | 56 size_t storage_offset in computeStorageNbytesContiguous()
78 size_t storage_offset in computeStorageNbytes()
127 const SymInt& storage_offset in computeStorageNbytesContiguous()
139 const SymInt& storage_offset in computeStorageNbytes()
|
| D | SparseTensorImpl.cpp | 65 void SparseTensorImpl::set_storage_offset(int64_t storage_offset) { in set_storage_offset()
|
| D | LegacyBatchedTensorImpl.cpp | 102 void BatchedTensorImpl::set_storage_offset(int64_t storage_offset) { in set_storage_offset()
|
| D | TensorGeometry.h | 65 int64_t storage_offset() const { in storage_offset() function
|
| D | SparseCsrTensorImpl.cpp | 250 void SparseCsrTensorImpl::set_storage_offset(int64_t storage_offset) { in set_storage_offset()
|
| /external/pytorch/aten/src/ATen/ops/ |
| D | from_blob.h | 34 TensorMaker& storage_offset(std::optional<int64_t> value) noexcept { in storage_offset() function
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | LegacyBatchingRegistrations.cpp | 289 c10::SymIntArrayRef sizes, c10::SymIntArrayRef strides, const c10::SymInt& storage_offset) { in maximum_indexable_location()
311 auto storage_offset = maybe_storage_offset.value_or(base_offset); in checkBasicAsStridedValidForSlice() local
364 std::optional<c10::SymInt> storage_offset) { in as_strided_batching_rule()
|
| /external/pytorch/torch/distributed/checkpoint/ |
| D | planner_helpers.py | 126 dest_index, dest_offset, storage_index, storage_offset, length argument
|
| /external/pytorch/aten/src/ATen/native/mps/ |
| D | TensorFactory.cpp | 143 Tensor& set_storage_mps_(Tensor& result, Storage storage, int64_t storage_offset, IntArrayRef size,… in set_storage_mps_()
|
| /external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/csrc/ |
| D | OpenRegMem.cpp | 109 int64_t storage_offset, in set_openreg()
|
| /external/pytorch/torch/fx/experimental/ |
| D | recording.py | 357 def storage_offset(self) -> Union[int, torch.SymInt]: member in FakeTensorMeta
|
| /external/webp/imageio/ |
| D | tiffdec.c | 31 size_t storage_offset; member
|