Copy.h 332 B

123456789101112131415
  1. // Copyright © 2022 Apple Inc.
  2. #pragma once
  3. #include <ATen/core/Tensor.h>
  4. namespace at {
  5. namespace native {
  6. namespace mps {
  7. at::Tensor& mps_copy_(at::Tensor& dst, const at::Tensor& src, bool non_blocking);
  8. void copy_blit_mps(void* dst, const void* src, size_t size);
  9. } // namespace mps
  10. } // namespace native
  11. } // namespace at