Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseCsrTensor.cpp408 auto compressed_indices_size = DimVector(size.slice(0, batch_dim)); in sparse_compressed_tensor_with_dims() local
707 auto compressed_indices_size = DimVector(size.slice(0, size.size() - 2)); in empty_sparse_compressed() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorFactories.cpp1285 auto compressed_indices_size = DimVector(size_.slice(0, size.size() - 2)); in zeros_sparse_compressed_symint() local