Searched defs:nested_strides (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | NestedTensorImpl.cpp | 18 const at::Tensor& nested_strides, in validate_nested_tensor_metadata()
170 at::Tensor nested_strides, in NestedTensorImpl()
194 at::Tensor nested_strides, in NestedTensorImpl()
227 at::Tensor nested_strides, in NestedTensorImpl()
|
| H A D | FunctionalInverses.cpp | 301 …seReturnMode inverse_return_mode, const Tensor& nested_sizes, const Tensor& nested_strides, const … in _nested_view_from_buffer_inverse()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorFactories.cpp | 64 auto nested_strides = self_nt->get_nested_strides().clone(); in empty_like_nested() local
213 auto nested_strides = nt_impl->get_nested_strides(); in narrow_nested_symint() local
237 const auto& nested_strides = nt_impl->get_nested_strides(); in alias_nested() local
|
| H A D | NestedTensorUtils.h | 50 at::Tensor nested_strides, in wrap_buffer()
85 at::Tensor nested_strides, in create_nested_view_tensor()
|
| H A D | NestedTensorMath.cpp | 875 const Tensor& nested_strides, in _nested_view_from_buffer()
|