Home
last modified time | relevance | path

Searched refs:CheckSparseTensorInvariants (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DSparseCsrTensorUtils.h143 class CheckSparseTensorInvariants {
147 CheckSparseTensorInvariants(bool state) { in CheckSparseTensorInvariants() function
152 ~CheckSparseTensorInvariants() { in ~CheckSparseTensorInvariants()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseTensor.cpp884 at::sparse_csr::CheckSparseTensorInvariants _(false); in _pin_memory_sparse_coo()
H A DSparseCsrTensor.cpp1272 CheckSparseTensorInvariants _(false); in _pin_memory_sparse_compressed()