File size: 271 Bytes
9dd3461 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
#pragma once
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
|