File size: 162 Bytes
9dd3461
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
#pragma once

namespace at {
struct TensorIteratorBase;

namespace native {

void direct_copy_kernel_cuda(TensorIteratorBase &iter);

}}  // namespace at::native