50 lines
873 B
C
50 lines
873 B
C
|
#pragma once
|
||
|
|
|
||
|
|
#include <c10/core/ScalarType.h>
|
||
|
|
|
||
|
|
namespace at::native {
|
||
|
|
|
||
|
|
void Unfold3dCopyCPU(
|
||
|
|
ScalarType dtype,
|
||
|
|
const void *src,
|
||
|
|
int64_t C,
|
||
|
|
int64_t X_D,
|
||
|
|
int64_t X_H,
|
||
|
|
int64_t X_W,
|
||
|
|
int64_t Y_D,
|
||
|
|
int64_t Y_H,
|
||
|
|
int64_t Y_W,
|
||
|
|
int64_t kernel_d,
|
||
|
|
int64_t kernel_h,
|
||
|
|
int64_t kernel_w,
|
||
|
|
int64_t stride_d,
|
||
|
|
int64_t stride_h,
|
||
|
|
int64_t stride_w,
|
||
|
|
int64_t pad_d,
|
||
|
|
int64_t pad_h,
|
||
|
|
int64_t pad_w,
|
||
|
|
void* dst);
|
||
|
|
|
||
|
|
void Unfold3dAccCPU(
|
||
|
|
ScalarType dtype,
|
||
|
|
const void *src,
|
||
|
|
int64_t C,
|
||
|
|
int64_t X_D,
|
||
|
|
int64_t X_H,
|
||
|
|
int64_t X_W,
|
||
|
|
int64_t Y_D,
|
||
|
|
int64_t Y_H,
|
||
|
|
int64_t Y_W,
|
||
|
|
int64_t kernel_d,
|
||
|
|
int64_t kernel_h,
|
||
|
|
int64_t kernel_w,
|
||
|
|
int64_t stride_d,
|
||
|
|
int64_t stride_h,
|
||
|
|
int64_t stride_w,
|
||
|
|
int64_t pad_d,
|
||
|
|
int64_t pad_h,
|
||
|
|
int64_t pad_w,
|
||
|
|
void *dst);
|
||
|
|
|
||
|
|
} // namespace at::native
|