1 #pragma once 2 3 #include <ATen/core/Tensor.h> 4 #include <ATen/native/DispatchStub.h> 5 #include <ATen/core/IListRef.h> 6 7 namespace at::native { 8 9 using cat_serial_fn = void(*)(const Tensor &, const MaterializedITensorListRef&, int64_t); 10 DECLARE_DISPATCH(cat_serial_fn, cat_serial_stub); 11 12 } // namespace at::native 13