1 #pragma once 2 #include <ATen/core/Tensor.h> 3 #include <ATen/SparseCsrTensorUtils.h> 4 5 namespace at { 6 namespace sparse_csr { 7 Tensor& _sparse_mm_mkl_( 8 Tensor& self, 9 const SparseCsrTensor& sparse_, 10 const Tensor& dense, 11 const Tensor& t, 12 const Scalar& alpha, 13 const Scalar& beta); 14 } // namespace native 15 } // namespace at 16