1 #pragma once 2 3 #include <ATen/native/DispatchStub.h> 4 5 namespace at { 6 7 class Tensor; 8 struct TensorIterator; 9 class TensorBase; 10 11 namespace native { 12 13 using copy_fn = void (*)(TensorIterator&, bool non_blocking); 14 15 DECLARE_DISPATCH(copy_fn, copy_stub); 16 17 TORCH_API void copy_ignoring_overlaps(const TensorBase &dst, const TensorBase &src); 18 19 } // namespace native 20 } // namespace at 21