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 |