1 #pragma once 2 3 #include <ATen/Tensor.h> 4 5 namespace at { 6 namespace native { 7 namespace sparse { 8 namespace impl { 9 namespace mkl { 10 11 void addmm_out_sparse_csr( 12 const Tensor& mat1, 13 const Tensor& mat2, 14 const Scalar& beta, 15 const Scalar& alpha, 16 const Tensor& result); 17 18 void addmv_out_sparse_csr( 19 const Tensor& mat, 20 const Tensor& vec, 21 const Scalar& beta, 22 const Scalar& alpha, 23 const Tensor& result); 24 25 void add_out_sparse_csr( 26 const Tensor& mat1, 27 const Tensor& mat2, 28 const Scalar& alpha, 29 const Tensor& result); 30 31 void triangular_solve_out_sparse_csr( 32 const Tensor& A, 33 const Tensor& B, 34 const Tensor& X, 35 bool upper, 36 bool transpose, 37 bool unitriangular); 38 39 } // namespace mkl 40 } // namespace impl 41 } // namespace sparse 42 } // namespace native 43 } // namespace at 44