1 #pragma once 2 3 #include <ATen/native/DispatchStub.h> 4 5 namespace at { 6 class Tensor; 7 8 namespace native { 9 10 using cross_fn = void(*)(const Tensor&, const Tensor&, const Tensor&, const int64_t d); 11 12 DECLARE_DISPATCH(cross_fn, cross_stub); 13 14 }} // namespace at::native 15