Home
last modified time | relevance | path

Searched refs:sizes_arrayref (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/c10/core/
H A DTensorImpl.cpp233 sizes_and_strides_.sizes_arrayref(), in compute_contiguous()
243 sizes_and_strides_.sizes_arrayref(), in compute_channels_last_contiguous_2d()
252 sizes_and_strides_.sizes_arrayref(), in compute_channels_last_contiguous_3d()
261 sizes_and_strides_.sizes_arrayref(), in compute_strides_like_channels_last_2d()
270 sizes_and_strides_.sizes_arrayref(), in compute_strides_like_channels_last_3d()
279 sizes_and_strides_.sizes_arrayref(), in compute_non_overlapping_and_dense()
659 IntArrayRef sizes_and_strides = sizes_and_strides_.sizes_arrayref(); in Extend()
719 IntArrayRef sizes_and_strides = sizes_and_strides_.sizes_arrayref(); in ReserveSpace()
H A DTensorImpl.h611 return sizes_and_strides_.sizes_arrayref(); in sizes()
620 sizes_and_strides_.sizes_arrayref()); in sym_sizes()
627 return sizes_and_strides_.sizes_arrayref(); in sizes_default()
2531 return c10::multiply_integers(sizes_and_strides_.sizes_arrayref()); in compute_numel()
2544 c10::safe_multiplies_u64(sizes_and_strides_.sizes_arrayref(), &n); in safe_compute_numel()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DOpaqueTensorImpl.h81 sizes_and_strides_.sizes_arrayref()); in shallow_copy_and_detach()
105 sizes_and_strides_.sizes_arrayref()); in shallow_copy_and_detach()
/aosp_15_r20/external/pytorch/c10/core/impl/
H A DSizesAndStrides.h151 IntArrayRef sizes_arrayref() const noexcept { in sizes_arrayref() function
/aosp_15_r20/external/pytorch/c10/test/core/impl/
H A DSizesAndStrides_test.cpp30 EXPECT_EQ(sz.sizes_arrayref(), sizes); in checkData()