1 #pragma once 2 3 #include <ATen/native/DispatchStub.h> 4 #include <cstdint> 5 6 namespace at { 7 struct TensorIterator; 8 9 namespace native { 10 11 using _compute_linear_combination_fn = void(*)( 12 TensorIterator& iter, 13 int64_t in_stride, 14 int64_t coeff_stride, 15 int64_t num_summations 16 ); 17 18 DECLARE_DISPATCH(_compute_linear_combination_fn, _compute_linear_combination_stub); 19 20 }} // namespace at::native 21