1 #pragma once 2 3 #include <ATen/native/DispatchStub.h> 4 5 namespace c10 { 6 class Scalar; 7 } 8 9 namespace at { 10 struct TensorIterator; 11 } 12 13 namespace at::native { 14 15 using addr_fn = void (*)(TensorIterator &, const Scalar& beta, const Scalar& alpha); 16 DECLARE_DISPATCH(addr_fn, addr_stub); 17 } // namespace at::native 18