Searched defs:nested_sizes (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | NestedTensorImpl.cpp | 17 const at::Tensor& nested_sizes, in validate_nested_tensor_metadata()
169 at::Tensor nested_sizes, in NestedTensorImpl()
193 at::Tensor nested_sizes, in NestedTensorImpl()
215 const at::Tensor& nested_sizes) in NestedTensorImpl()
226 at::Tensor nested_sizes, in NestedTensorImpl()
|
| H A D | FunctionalInverses.cpp | 301 …t Tensor& mutated_view, InverseReturnMode inverse_return_mode, const Tensor& nested_sizes, const T… in _nested_view_from_buffer_inverse()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorUtils.h | 34 inline at::Tensor wrap_buffer(at::Tensor buffer, at::Tensor nested_sizes) { in wrap_buffer()
49 at::Tensor nested_sizes, in wrap_buffer()
84 at::Tensor nested_sizes, in create_nested_view_tensor()
|
| H A D | NestedTensorFactories.cpp | 212 auto nested_sizes = nt_impl->get_nested_sizes(); in narrow_nested_symint() local
236 const auto& nested_sizes = nt_impl->get_nested_sizes(); in alias_nested() local
|
| H A D | NestedTensorMath.cpp | 874 const Tensor& nested_sizes, in _nested_view_from_buffer()
|