Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DEmbeddingBackwardKernel.cu93 const int startFeature = gid % stride_warped; in compute_grad_weight_bags() local
139 const int startFeature = gid % stride_warped; in compute_grad_weight() local
172 const int startFeature = gid % stride_warped; in sum_and_scatter() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseCUDAApplyUtils.cuh210 const int startFeature = threadIdx.x + blockIdx.y * blockDim.x * SZ; in C10_LAUNCH_BOUNDS_1() local
260 const int startFeature = threadIdx.x + blockIdx.y * blockDim.x * SZ; in coalesceValuesKernel() local