Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseCsrTensor.cpp784 … auto self_blocksize = DimVector(self_values.sizes().slice(self_values.dim()-2, 2)); in copy_sparse_compressed_() local
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DFunctionsManual.cpp7010 const c10::OptionalArrayRef<c10::SymInt>& self_blocksize) { in to_sparse_backward()