1 #pragma once 2 3 #include <c10/core/ScalarType.h> 4 5 namespace at::native { 6 7 void Unfold3dCopyCPU( 8 ScalarType dtype, 9 const void *src, 10 int64_t C, 11 int64_t X_D, 12 int64_t X_H, 13 int64_t X_W, 14 int64_t Y_D, 15 int64_t Y_H, 16 int64_t Y_W, 17 int64_t kernel_d, 18 int64_t kernel_h, 19 int64_t kernel_w, 20 int64_t stride_d, 21 int64_t stride_h, 22 int64_t stride_w, 23 int64_t pad_d, 24 int64_t pad_h, 25 int64_t pad_w, 26 void* dst); 27 28 void Unfold3dAccCPU( 29 ScalarType dtype, 30 const void *src, 31 int64_t C, 32 int64_t X_D, 33 int64_t X_H, 34 int64_t X_W, 35 int64_t Y_D, 36 int64_t Y_H, 37 int64_t Y_W, 38 int64_t kernel_d, 39 int64_t kernel_h, 40 int64_t kernel_w, 41 int64_t stride_d, 42 int64_t stride_h, 43 int64_t stride_w, 44 int64_t pad_d, 45 int64_t pad_h, 46 int64_t pad_w, 47 void *dst); 48 49 } // namespace at::native 50