Home
last modified time | relevance | path

Searched refs:checkSparseTensorInvariants (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_new.cpp876 : state{at::globalContext().checkSparseTensorInvariants()} {} in CheckSparseTensorInvariantsContext()
946 at::globalContext().checkSparseTensorInvariants(); in sparse_compressed_tensor_ctor_worker()
1173 at::globalContext().checkSparseTensorInvariants(); in sparse_coo_tensor_ctor()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DContext.cpp438 bool Context::checkSparseTensorInvariants() const { in checkSparseTensorInvariants() function in at::Context
H A DSparseCsrTensorUtils.h148 old_state = at::globalContext().checkSparseTensorInvariants(); in CheckSparseTensorInvariants()
H A DContext.h322 bool checkSparseTensorInvariants() const;
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseCsrTensor.cpp466 if (at::globalContext().checkSparseTensorInvariants()) { in _sparse_compressed_tensor_unsafe_symint()
490 if (at::globalContext().checkSparseTensorInvariants()) { in _sparse_compressed_tensor_unsafe_template()
H A DSparseTensor.cpp483 if (at::globalContext().checkSparseTensorInvariants()) { in _sparse_coo_tensor_unsafe()
/aosp_15_r20/external/pytorch/torch/csrc/
H A DModule.cpp1171 if (at::globalContext().checkSparseTensorInvariants()) in THPModule_checkSparseTensorInvariants()