Home
last modified time | relevance | path

Searched full:nested_strides (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DNestedTensorImpl.cpp18 const at::Tensor& nested_strides, in validate_nested_tensor_metadata() argument
23 TORCH_INTERNAL_ASSERT(nested_strides.is_contiguous()); in validate_nested_tensor_metadata()
24 TORCH_INTERNAL_ASSERT(nested_strides.dim() == size_dim); in validate_nested_tensor_metadata()
25 TORCH_INTERNAL_ASSERT(nested_sizes.sizes() == nested_strides.sizes()); in validate_nested_tensor_metadata()
170 at::Tensor nested_strides, in NestedTensorImpl() argument
174 nested_strides_(std::move(nested_strides)), in NestedTensorImpl()
194 at::Tensor nested_strides, in NestedTensorImpl() argument
201 std::move(nested_strides), in NestedTensorImpl()
211 // assume contiguous, `nested_strides` and `offsets`
227 at::Tensor nested_strides, in NestedTensorImpl() argument
[all …]
H A DNestedTensorImpl.h26 at::Tensor nested_strides,
32 at::Tensor nested_strides,
34 // assume contiguous, `nested_strides` and `offsets`
45 at::Tensor nested_strides,
93 * nested_strides and storage_offsets.
H A DFunctionalInverses.cpp301 …nMode inverse_return_mode, const Tensor& nested_sizes, const Tensor& nested_strides, const Tensor&… in _nested_view_from_buffer_inverse() argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorFactories.cpp64 auto nested_strides = self_nt->get_nested_strides().clone(); in empty_like_nested() local
66 auto tensor = wrap_buffer(new_buffer, nested_size, nested_strides, offsets); in empty_like_nested()
146 // efficient implementation of nested_sizes_ and nested_strides. in clone_nested()
213 auto nested_strides = nt_impl->get_nested_strides(); in narrow_nested_symint() local
222 nested_strides = nested_strides.narrow(0, start_int, length_int); in narrow_nested_symint()
229 nested_strides, in narrow_nested_symint()
237 const auto& nested_strides = nt_impl->get_nested_strides(); in alias_nested() local
243 nested_strides, in alias_nested()
H A DNestedTensorUtils.h50 at::Tensor nested_strides, in wrap_buffer() argument
57 std::move(nested_strides), in wrap_buffer()
78 * @param nested_strides View tensors' strides.
85 at::Tensor nested_strides, in create_nested_view_tensor() argument
98 nested_strides, in create_nested_view_tensor()
H A DNestedTensorMath.cpp875 const Tensor& nested_strides, in _nested_view_from_buffer() argument
892 …TORCH_INTERNAL_ASSERT(nested_strides.dim() == 2, "Expected the nested stride tensor to be two dime… in _nested_view_from_buffer()
893 …TORCH_INTERNAL_ASSERT(nested_sizes.size(0) == nested_strides.size(0), "Expected the first dimensio… in _nested_view_from_buffer()
894 …TORCH_INTERNAL_ASSERT(nested_strides.size(0) == storage_offsets.size(0), "Expected the first dimen… in _nested_view_from_buffer()
899 nested_strides, in _nested_view_from_buffer()
/aosp_15_r20/external/pytorch/tools/autograd/
H A Dderivatives.yaml2792 - name: _nested_view_from_buffer(Tensor(a) self, Tensor nested_size, Tensor nested_strides, Tensor…
2795 nested_strides: non_differentiable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dnative_functions.yaml6174 - func: _nested_view_from_buffer(Tensor(a) self, Tensor nested_size, Tensor nested_strides, Tensor …
6180 - func: _nested_view_from_buffer_copy(Tensor self, Tensor nested_size, Tensor nested_strides, Tenso…