blob: 6f9fa98446ac88605f48776eb504f9745456716f [file] [log] [blame]
#ifndef THC_GENERIC_FILE
#define THC_GENERIC_FILE "generic/SpatialFullConvolution.cu"
#else
void THNN_(SpatialFullConvolution_updateOutput)(
THCState *state,
THCTensor *input,
THCTensor *output,
THCTensor *weight,
THCTensor *bias,
THCTensor *columns,
THCTensor *ones,
int kW, int kH,
int dW, int dH,
int padW, int padH,
int adjW, int adjH)
{
THNN_(SpatialFullDilatedConvolution_updateOutput)(
state, input, output, weight, bias, columns, ones,
kW, kH, dW, dH, padW, padH, 1, 1, adjW, adjH);
}
void THNN_(SpatialFullConvolution_updateGradInput)(
THCState *state,
THCTensor *input,
THCTensor *gradOutput,
THCTensor *gradInput,
THCTensor *weight,
THCTensor *gradColumns,
int kW, int kH,
int dW, int dH,
int padW, int padH,
int adjW, int adjH)
{
THNN_(SpatialFullDilatedConvolution_updateGradInput)(
state, input, gradOutput, gradInput, weight, gradColumns,
kW, kH, dW, dH, padW, padH, 1, 1, adjW, adjH);
}
void THNN_(SpatialFullConvolution_accGradParameters)(
THCState *state,
THCTensor *input,
THCTensor *gradOutput,
THCTensor *gradWeight,
THCTensor *gradBias,
THCTensor *columns,
THCTensor *ones,
int kW, int kH,
int dW, int dH,
int padW, int padH,
int adjW, int adjH,
accreal scale_)
{
THNN_(SpatialFullDilatedConvolution_accGradParameters)(
state, input, gradOutput, gradWeight, gradBias,
columns, ones,
kW, kH, dW, dH, padW, padH, 1, 1, adjW, adjH, scale_);
}
#endif