Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/detail/
H A DTensorInfo.cuh86 IndexType curDimIndex = linearId % info.sizes[i]; in get() local
106 IndexType curDimIndex = linearId % info.sizes[i]; in get() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DIndexingUtils.cpp21 auto curDimIndex = linearId % t.sym_size(i); in canUse32BitIndexMath() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DShape.cu119 IndexType curDimIndex = linearIndex - curDimSize * nextDimIndex; in compute() local