Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_new.h52 void _validate_sparse_coo_tensor_args(
H A Dtensor_new.cpp1261 void _validate_sparse_coo_tensor_args( in _validate_sparse_coo_tensor_args() function
1290 at::native::_validate_sparse_coo_tensor_args(indices, values, r.intlist(2)); in _validate_sparse_coo_tensor_args()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseTensor.cpp369 void _validate_sparse_coo_tensor_args( in _validate_sparse_coo_tensor_args() function
484 at::native::_validate_sparse_coo_tensor_args(indices, values_, size, is_coalesced); in _sparse_coo_tensor_unsafe()
/aosp_15_r20/external/pytorch/torch/
H A D_utils.py254 torch._validate_sparse_coo_tensor_args(
/aosp_15_r20/external/pytorch/test/
H A Dtest_sparse.py3740 … torch._validate_sparse_coo_tensor_args(s_res._indices(), s_res._values(), s_res.shape)
4931 torch._validate_sparse_coo_tensor_args(r._indices(), r._values(), r.shape)
5213 … torch._validate_sparse_coo_tensor_args(result._indices(), result._values(), result.shape)