CopyKernel.h 271 B

123456789101112
  1. #pragma once
  2. namespace at {
  3. struct TensorIteratorBase;
  4. namespace native {
  5. inline namespace CPU_CAPABILITY {
  6. void direct_copy_kernel(TensorIteratorBase &iter);
  7. void copy_kernel(TensorIterator& iter, bool /*non_blocking*/);
  8. }}} // namespace at::native::CPU_CAPABILITY