Copy.h 373 B

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