File size: 326 Bytes
c1af2fa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
#pragma once
#include <ATen/native/TensorIterator.h>
namespace at {
struct TensorIteratorBase;
namespace native {
inline namespace CPU_CAPABILITY {
void direct_copy_kernel(TensorIteratorBase &iter);
void copy_kernel(TensorIterator& iter, bool /*non_blocking*/);
}}} // namespace at::native::CPU_CAPABILITY
|