Searched refs:nested_strides (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/aten/src/ATen/ |
| D | NestedTensorImpl.cpp | 18 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()
227 at::Tensor nested_strides, in NestedTensorImpl() argument
231 nested_strides_(std::move(nested_strides)), in NestedTensorImpl()
|
| D | NestedTensorImpl.h | 26 at::Tensor nested_strides,
32 at::Tensor nested_strides,
45 at::Tensor nested_strides,
|
| D | FunctionalInverses.cpp | 301 …nMode inverse_return_mode, const Tensor& nested_sizes, const Tensor& nested_strides, const Tensor&… in _nested_view_from_buffer_inverse() argument
|
| /external/pytorch/aten/src/ATen/native/nested/ |
| D | NestedTensorFactories.cpp | 64 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()
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()
|
| D | NestedTensorUtils.h | 50 at::Tensor nested_strides, in wrap_buffer() argument
57 std::move(nested_strides), in wrap_buffer()
85 at::Tensor nested_strides, in create_nested_view_tensor() argument
98 nested_strides, in create_nested_view_tensor()
|
| D | NestedTensorMath.cpp | 875 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()
|