Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/sparse/cuda/
DSparseCUDAApplyUtils.cuh114 int64_t nDimI = r_indices.sizes[0]; in C10_LAUNCH_BOUNDS_2() local
156 int64_t nDimI = r_indices.sizes[0]; in C10_LAUNCH_BOUNDS_2() local
DSparseCUDATensorMath.cu306 int64_t nDimI = sparse.sparse_dim(); in add_out_dense_sparse_cuda() local