Home
last modified time | relevance | path

Searched refs:setCheckSparseTensorInvariants (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_new.cpp878 at::globalContext().setCheckSparseTensorInvariants(state); in ~CheckSparseTensorInvariantsContext()
958 at::globalContext().setCheckSparseTensorInvariants( in sparse_compressed_tensor_ctor_worker()
1014 at::globalContext().setCheckSparseTensorInvariants( in sparse_compressed_tensor_ctor_worker()
1183 at::globalContext().setCheckSparseTensorInvariants( in sparse_coo_tensor_ctor()
1217 at::globalContext().setCheckSparseTensorInvariants( in sparse_coo_tensor_ctor()
1250 at::globalContext().setCheckSparseTensorInvariants( in sparse_coo_tensor_ctor()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DSparseCsrTensorUtils.h149 at::globalContext().setCheckSparseTensorInvariants(state); in CheckSparseTensorInvariants()
153 at::globalContext().setCheckSparseTensorInvariants(old_state); in ~CheckSparseTensorInvariants()
H A DContext.cpp434 void Context::setCheckSparseTensorInvariants(bool e) { in setCheckSparseTensorInvariants() function in at::Context
H A DContext.h321 void setCheckSparseTensorInvariants(bool e);
/aosp_15_r20/external/pytorch/torch/csrc/
H A DModule.cpp1163 at::globalContext().setCheckSparseTensorInvariants(arg == Py_True); in THPModule_setCheckSparseTensorInvariants()