1 #pragma once 2 #include <ATen/native/DispatchStub.h> 3 4 namespace at { 5 struct TensorIterator; 6 class TensorBase; 7 8 namespace native { 9 10 using spdiags_kernel_fn_t = 11 void (*)(TensorIterator&, const TensorBase&, TensorBase&, TensorBase&); 12 13 DECLARE_DISPATCH(spdiags_kernel_fn_t, spdiags_kernel_stub); 14 } // namespace native 15 } // namespace at 16