/usr/local/lib64/python3.6/site-packages/torch/include/ATen
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/ATen/SparseCsrTensorUtils.h (523B)
#pragma once
#include
#include
#include
#include
namespace at {
namespace sparse_csr {
using SparseCsrTensor = Tensor;
inline SparseCsrTensorImpl* get_sparse_csr_impl(const SparseCsrTensor& self) {
AT_ASSERTM(
self.is_sparse_csr(),
"_internal_get_SparseCsrTensorImpl: not a sparse CSR tensor");
return static_cast(self.unsafeGetTensorImpl());
}
} // namespace sparse
} // namespace at