1 #pragma once 2 3 #include <c10/macros/Export.h> 4 5 #include <ATen/Tensor.h> 6 #include <ATen/core/Scalar.h> 7 8 namespace at::native::sparse { 9 10 TORCH_API void sparse_sampled_addmm_check_inputs( 11 const Tensor& self, 12 const Tensor& mat1, 13 const Tensor& mat2, 14 const Scalar& beta, 15 const Scalar& alpha, 16 const Tensor& result); 17 18 } // namespace at::native::sparse 19