Home
last modified time | relevance | path

Searched defs:storage_offset (Results 1 – 25 of 50) sorted by relevance

12

/aosp_15_r20/external/executorch/kernels/test/
H A Dop_as_strided_copy_test.cpp34 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
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DResize.h85 T storage_offset, in checkInBoundsForStorage()
114 inline void checkSetStorage(Tensor& result, Storage storage, T storage_offset, in checkSetStorage()
156 T storage_offset) { in setStrided()
H A DTensorShape.cpp379 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 …]
H A DFill.cpp112 int64_t storage_offset = self.storage_offset(); in fill_diagonal_() local
/aosp_15_r20/external/pytorch/torch/
H A D_utils.py172 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
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DEmptyTensor.cpp56 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()
H A DSparseTensorImpl.cpp65 void SparseTensorImpl::set_storage_offset(int64_t storage_offset) { in set_storage_offset()
H A DLegacyBatchingRegistrations.cpp512 IntArrayRef sizes, IntArrayRef strides, int64_t storage_offset) { in maximum_indexable_location()
534 auto storage_offset = maybe_storage_offset.value_or(base_offset); in checkBasicAsStridedValidForSlice() local
617 std::optional<int64_t> storage_offset) { in as_strided_batching_rule()
H A DLegacyBatchedTensorImpl.cpp102 void BatchedTensorImpl::set_storage_offset(int64_t storage_offset) { in set_storage_offset()
H A DTensorGeometry.h65 int64_t storage_offset() const { in storage_offset() function
H A DSparseCsrTensorImpl.cpp250 void SparseCsrTensorImpl::set_storage_offset(int64_t storage_offset) { in set_storage_offset()
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_runtime/
H A Dutils.h108 int64_t storage_offset() { in storage_offset() function
109 int64_t storage_offset = 0; in storage_offset() local
H A Darrayref_tensor.h283 int64_t storage_offset) { in reinterpret_tensor_wrapper()
296 int64_t storage_offset) { in reinterpret_tensor_wrapper()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DLegacyBatchingRegistrations.cpp289 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()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DResize.h37 const auto storage_offset = self->storage_offset(); in resize_impl_cuda_() local
H A DTensorShapeCUDA.cpp31 Tensor& set_storage_cuda_(Tensor& result, Storage storage, int64_t storage_offset, IntArrayRef size… in set_storage_cuda_()
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_as_strided_copy.cpp24 optional<int64_t> storage_offset, in as_strided_copy_out()
H A Dop_diagonal_copy.cpp30 int64_t storage_offset = 0; in diagonal_copy_impl() local
/aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt/
H A Das_strided_test.py37 def init(self, M, N, size, stride, storage_offset, device): argument
/aosp_15_r20/external/pytorch/aten/src/ATen/ops/
H A Dfrom_blob.h34 TensorMaker& storage_offset(std::optional<int64_t> value) noexcept { in storage_offset() function
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dshape_inference.cpp1190 const int64_t& storage_offset) { in compute_shape_as_strided_view_update()
1197 const int64_t& storage_offset) { in compute_shape_as_strided()
1350 ::std::optional<c10::SymInt> storage_offset) { in compute_shape_as_strided_scatter_symint()
/aosp_15_r20/external/pytorch/torch/distributed/checkpoint/
H A Dplanner_helpers.py126 dest_index, dest_offset, storage_index, storage_offset, length argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/
H A DTensorFactory.cpp143 Tensor& set_storage_mps_(Tensor& result, Storage storage, int64_t storage_offset, IntArrayRef size,… in set_storage_mps_()
/aosp_15_r20/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/csrc/
H A DOpenRegMem.cpp109 int64_t storage_offset, in set_openreg()
/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Drecording.py357 def storage_offset(self) -> Union[int, torch.SymInt]: member in FakeTensorMeta

12