Home
last modified time | relevance | path

Searched refs:unsafe_mutable_descriptor (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDASparseDescriptors.h141 cusparseDnMatDescr* unsafe_mutable_descriptor() const { in unsafe_mutable_descriptor() function
144 cusparseDnMatDescr* unsafe_mutable_descriptor() { in unsafe_mutable_descriptor() function
177 cusparseDnMatDescr* unsafe_mutable_descriptor() const { in unsafe_mutable_descriptor() function
180 cusparseDnMatDescr* unsafe_mutable_descriptor() { in unsafe_mutable_descriptor() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseBlasImpl.cpp658 descB.unsafe_mutable_descriptor(), in spmm()
675 descB.unsafe_mutable_descriptor(), in spmm()
1463 descA.unsafe_mutable_descriptor(), in sampled_addmm_out_sparse_csr()
1464 descB.unsafe_mutable_descriptor(), in sampled_addmm_out_sparse_csr()
1480 descA.unsafe_mutable_descriptor(), in sampled_addmm_out_sparse_csr()
1481 descB.unsafe_mutable_descriptor(), in sampled_addmm_out_sparse_csr()
1493 descA.unsafe_mutable_descriptor(), in sampled_addmm_out_sparse_csr()
1494 descB.unsafe_mutable_descriptor(), in sampled_addmm_out_sparse_csr()