Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/
DTensorShape.cpp1274 Tensor nzIndices = mask.nonzero().view(-1); in narrow_copy_sparse() local
1745 auto nzIndices = (indices[dim] == index).nonzero().view(-1); in select_sparse() local