Home
last modified time | relevance | path

Searched defs:self_j (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DDistanceOpsKernel.cpp170 const scalar_t * self_j = self_start + j * m; in run_parallel_pdist() local
230 const scalar_t * self_j = t2_start + size2 * l + j; in run_parallel_cdist() local
276 const scalar_t * self_j = self_i + m; in backward_down_column_pdist() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DDistanceKernel.cu144 const scalar_t * self_j = x2 + l * l2_size + j * m + init; in cdist_backward_kernel_cuda_impl() local
177 const scalar_t * self_j = self + j * m + init; in pdist_backward_kernel_cuda_impl() local