Home
last modified time | relevance | path

Searched defs:inner_strides (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DCUDALoops.cuh307 auto inner_strides = iter.get_inner_strides(); in gpu_kernel_impl() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DSumKernel.cpp566 int64_t inner_strides[3] = { strides[0], strides[0], strides[1] }; in cascade_sum() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DTensorIterator.cpp700 auto inner_strides = StrideVector(); in get_dim_strides() local