Learn more »
Push, build, and install
RubyGems
npm packages
Python packages
Maven artifacts
PHP packages
Go Modules
Bower components
Debian packages
RPM packages
NuGet packages
#pragma once
#include <ATen/ATen.h>
#include <ATen/native/DispatchStub.h>
namespace at { namespace native {
using unfold2d_fn =
void (*)(
Tensor& finput,
Tensor& input,
int64_t kH,
int64_t kW,
int64_t dH,
int64_t dW,
int64_t padH,
int64_t padW,
int64_t n_input_plane,
int64_t input_height,
int64_t input_width,
int64_t output_height,
int64_t output_width
);
DECLARE_DISPATCH(unfold2d_fn, unfolded2d_copy_stub);
DECLARE_DISPATCH(unfold2d_fn, unfolded2d_acc_stub);
}} // namespace at::native