Home
last modified time | relevance | path

Searched defs:self_strides (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_constant_pad_nd.cpp35 IntArrayRef self_strides, in apply_padding_to_dim()
120 int64_t self_strides[kTensorDimensionLimit]; in constant_pad_nd_out_impl() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DScatterGatherKernel.cu170 auto self_strides = ensure_nonempty_vec(self.strides().vec()); in operator ()() local
228 auto self_strides = ensure_nonempty_vec(self.strides().vec()); in operator ()() local
287 auto self_strides = ensure_nonempty_vec(self.strides().vec()); in operator ()() local
H A DSpectralOps.cpp183 auto self_strides = self.strides(); in _exec_fft() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkl/
H A DSpectralOps.cpp447 auto self_strides = self.strides(); in _exec_fft() local
510 auto self_strides = self.strides(); in _sort_dims() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorMatmul.cpp83 std::vector<IntArrayRef> self_strides = NestedTensor_get_strides(self_ptr); in matmul_with_bmm_nested() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorAdvancedIndexing.cpp1528 auto self_strides = self_nonzero_dim.strides().vec(); in index_fill_() local