mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/18959 ghimport-source-id: a934163fa34cb2019732d5f49dc7290c376bf156 Differential Revision: D14831246 Pulled By: ezyang fbshipit-source-id: beb92dc4ee8c82f4c8259c081dd72e477fe7a9d0
1408 lines
50 KiB
C
1408 lines
50 KiB
C
#ifndef THC_GENERIC_FILE
|
|
#define THC_GENERIC_FILE "THCUNN/generic/THCUNN.h"
|
|
#else
|
|
|
|
#include <ATen/core/Reduction.h>
|
|
|
|
THC_API void THNN_(Abs_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output);
|
|
|
|
THC_API void THNN_(Abs_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput);
|
|
|
|
THC_API void THNN_(AbsCriterion_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *target,
|
|
THCTensor *output,
|
|
int64_t reduction);
|
|
|
|
THC_API void THNN_(AbsCriterion_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *target,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int64_t reduction);
|
|
|
|
THC_API void THNN_(BCECriterion_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *target,
|
|
THCTensor *output,
|
|
int64_t reduction,
|
|
THCTensor *weights); // [OPTIONAL]
|
|
|
|
THC_API void THNN_(BCECriterion_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *target,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int64_t reduction,
|
|
THCTensor *weights); // [OPTIONAL]
|
|
|
|
THC_API void THNN_(ClassNLLCriterion_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCIndexTensor *target,
|
|
THCTensor *output,
|
|
int64_t reduction,
|
|
THCTensor *weights, // [OPTIONAL]
|
|
THCTensor *total_weight,
|
|
int64_t ignore_index);
|
|
|
|
THC_API void THNN_(ClassNLLCriterion_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCIndexTensor *target,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int64_t reduction,
|
|
THCTensor *weights, // [OPTIONAL]
|
|
THCTensor *total_weight,
|
|
int64_t ignore_index);
|
|
|
|
THC_API void THNN_(DistKLDivCriterion_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *target,
|
|
THCTensor *output,
|
|
int64_t reduction);
|
|
|
|
THC_API void THNN_(DistKLDivCriterion_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *target,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int64_t reduction);
|
|
|
|
THC_API void THNN_(ELU_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
accreal alpha,
|
|
accreal scale,
|
|
accreal input_scale,
|
|
bool inplace);
|
|
|
|
THC_API void THNN_(ELU_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *output,
|
|
accreal alpha,
|
|
accreal scale,
|
|
accreal input_scale);
|
|
|
|
THC_API void THNN_(FeatureLPPooling_updateOutput)(
|
|
THCState* state,
|
|
THCTensor* inputTH,
|
|
THCTensor* outputTH,
|
|
accreal power,
|
|
int width,
|
|
int stride,
|
|
bool batchMode);
|
|
|
|
THC_API void THNN_(FeatureLPPooling_updateGradInput)(
|
|
THCState* state,
|
|
THCTensor* gradOutputTH,
|
|
THCTensor* inputTH,
|
|
THCTensor* outputTH,
|
|
THCTensor* gradInputTH,
|
|
accreal power,
|
|
int width,
|
|
int stride,
|
|
bool batchMode);
|
|
|
|
THC_API void THNN_(HardTanh_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
accreal min_val,
|
|
accreal max_val,
|
|
bool inplace);
|
|
|
|
THC_API void THNN_(HardTanh_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
accreal min_val,
|
|
accreal max_val,
|
|
bool inplace);
|
|
|
|
THC_API void THNN_(GatedLinear_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
int dim);
|
|
|
|
THC_API void THNN_(GatedLinear_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int dim);
|
|
|
|
THC_API void THNN_(Im2Col_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
int64_t kH, int64_t kW,
|
|
int64_t dH, int64_t dW,
|
|
int64_t padH, int64_t padW,
|
|
int64_t sH, int64_t sW);
|
|
|
|
THC_API void THNN_(Im2Col_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int64_t inputHeight, int64_t inputWidth,
|
|
int64_t kH, int64_t kW,
|
|
int64_t dH, int64_t dW,
|
|
int64_t padH, int64_t padW,
|
|
int64_t sH, int64_t sW);
|
|
|
|
THC_API void THNN_(Col2Im_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
int64_t outputHeight, int64_t outputWidth,
|
|
int64_t kH, int64_t kW,
|
|
int64_t dH, int64_t dW,
|
|
int64_t padH, int64_t padW,
|
|
int64_t sH, int64_t sW);
|
|
|
|
THC_API void THNN_(Col2Im_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int64_t kH, int64_t kW,
|
|
int64_t dH, int64_t dW,
|
|
int64_t padH, int64_t padW,
|
|
int64_t sH, int64_t sW);
|
|
|
|
THC_API void THNN_(LeakyReLU_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
accreal negval,
|
|
bool inplace);
|
|
|
|
THC_API void THNN_(LeakyReLU_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
accreal negval,
|
|
bool inplace);
|
|
|
|
THC_API void THNN_(LogSigmoid_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCTensor *buffer);
|
|
|
|
THC_API void THNN_(LogSigmoid_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *buffer);
|
|
|
|
THC_API void THNN_(LookupTable_accGradParameters)(
|
|
THCState *state,
|
|
THCIndexTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradWeight,
|
|
THCIndexTensor *count,
|
|
THCIndexTensor *sorted, // [OPTIONAL]
|
|
THCIndexTensor *indices, // [OPTIONAL]
|
|
bool scaleGradByFreq,
|
|
int paddingValue,
|
|
accreal scale);
|
|
|
|
THC_API void THNN_(LookupTable_renorm)(
|
|
THCState *state,
|
|
THCIndexTensor *idx,
|
|
THCTensor *weight,
|
|
accreal maxNorm,
|
|
accreal normType);
|
|
|
|
THC_API void THNN_(LookupTableBag_updateOutput)(
|
|
THCState *state,
|
|
THCIndexTensor *input,
|
|
THCIndexTensor *offsets,
|
|
THCTensor *weight,
|
|
THCTensor *output,
|
|
THCIndexTensor *offset2bag,
|
|
int mode,
|
|
THCIndexTensor *seq_length); // [OPTIONAL]
|
|
|
|
THC_API void THNN_(LookupTableBag_accGradParameters)(
|
|
THCState *state,
|
|
THCIndexTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradWeight,
|
|
THCIndexTensor *offset2bag,
|
|
THCIndexTensor *count,
|
|
THCIndexTensor *sortedIndices,
|
|
THCIndexTensor *origIndices,
|
|
bool scaleGradByFreq,
|
|
int mode,
|
|
THCIndexTensor *seq_length, // [OPTIONAL]
|
|
accreal scale_);
|
|
|
|
THC_API void THNN_(L1Cost_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output);
|
|
|
|
THC_API void THNN_(L1Cost_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput, // [OPTIONAL]
|
|
THCTensor *gradInput);
|
|
|
|
THC_API void THNN_(MarginCriterion_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *target,
|
|
THCTensor *output,
|
|
bool sizeAverage,
|
|
accreal margin);
|
|
|
|
THC_API void THNN_(MarginCriterion_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *target,
|
|
THCTensor *gradInput,
|
|
bool sizeAverage,
|
|
accreal margin);
|
|
|
|
THC_API void THNN_(MSECriterion_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *target,
|
|
THCTensor *output,
|
|
int64_t reduction);
|
|
|
|
THC_API void THNN_(MSECriterion_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *target,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int64_t reduction);
|
|
|
|
THC_API void THNN_(MultiLabelMarginCriterion_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCIndexTensor *target,
|
|
THCTensor *output,
|
|
THCTensor *istarget,
|
|
int64_t reduction);
|
|
|
|
THC_API void THNN_(MultiLabelMarginCriterion_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCIndexTensor *target,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *istarget,
|
|
int64_t reduction);
|
|
|
|
THC_API void THNN_(MultiMarginCriterion_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCIndexTensor *target,
|
|
THCTensor *output,
|
|
int64_t reduction,
|
|
int p,
|
|
THCTensor *weights, // [OPTIONAL]
|
|
accreal margin);
|
|
|
|
THC_API void THNN_(MultiMarginCriterion_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCIndexTensor *target,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int64_t reduction,
|
|
int p,
|
|
THCTensor *weights, // [OPTIONAL]
|
|
accreal margin);
|
|
THC_API void THNN_(SmoothL1Criterion_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *target,
|
|
THCTensor *output,
|
|
int64_t reduction);
|
|
|
|
THC_API void THNN_(SmoothL1Criterion_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *target,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int64_t reduction);
|
|
|
|
THC_API void THNN_(IndexLinear_updateOutput)(
|
|
THCState *state,
|
|
THCIndexTensor *keys,
|
|
int64_t keysOffset,
|
|
THCTensor *values,
|
|
THCIndexTensor *sizes,
|
|
THCIndexTensor *cumSumSizes,
|
|
THCTensor *output,
|
|
THCTensor *weight,
|
|
THCTensor *bias,
|
|
THCTensor *normalizedValues,
|
|
int train);
|
|
|
|
THC_API void THNN_(IndexLinear_accGradParameters)(
|
|
THCState *state,
|
|
THCIndexTensor *keys,
|
|
int64_t keysOffset,
|
|
THCTensor *values,
|
|
THCIndexTensor *sizes,
|
|
THCIndexTensor *cumSumSizes,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradWeight,
|
|
THCTensor *gradBias,
|
|
THCTensor *weight,
|
|
THCTensor *bias,
|
|
THCTensor* valuesBuffer,
|
|
accreal weightDecay,
|
|
accreal scale);
|
|
|
|
THC_API void THNN_(IndexLinear_accUpdateGradParameters)(
|
|
THCState *state,
|
|
THCIndexTensor *keys,
|
|
int64_t keysOffset,
|
|
THCTensor *values,
|
|
THCIndexTensor *sizes,
|
|
THCIndexTensor *cumSumSizes,
|
|
THCTensor *gradOutput,
|
|
THCTensor *weight,
|
|
THCTensor *bias,
|
|
accreal weightDecay,
|
|
accreal scale);
|
|
|
|
THC_API void THNN_(IndexLinear_updateParameters)(
|
|
THCState *state,
|
|
THCTensor *gradWeight,
|
|
THCTensor *gradBias,
|
|
THCTensor *weight,
|
|
THCTensor *bias,
|
|
THCIndexTensor *runningKeys,
|
|
THCIndexTensor *cumSumSizes,
|
|
int64_t keysOffset,
|
|
accreal weightDecay,
|
|
accreal learningRate);
|
|
|
|
THC_API void THNN_(SpatialAdaptiveMaxPooling_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCIndexTensor *indices,
|
|
int osizeW,
|
|
int osizeH);
|
|
|
|
THC_API void THNN_(SpatialAdaptiveMaxPooling_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCIndexTensor *indices);
|
|
|
|
THC_API void THNN_(SpatialAveragePooling_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
bool ceil_mode,
|
|
bool count_include_pad);
|
|
|
|
THC_API void THNN_(SpatialAveragePooling_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
bool ceil_mode,
|
|
bool count_include_pad);
|
|
|
|
THC_API void THNN_(SpatialClassNLLCriterion_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCIndexTensor *target,
|
|
THCTensor *output,
|
|
int64_t reduction,
|
|
THCTensor *weights, // [OPTIONAL]
|
|
THCTensor *total_weight,
|
|
int64_t ignore_index);
|
|
|
|
THC_API void THNN_(SpatialClassNLLCriterion_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCIndexTensor *target,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int64_t reduction,
|
|
THCTensor *weights, // [OPTIONAL]
|
|
THCTensor *total_weight,
|
|
int64_t ignore_index);
|
|
|
|
THC_API void THNN_(SpatialConvolutionLocal_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCTensor *weight,
|
|
THCTensor *bias,
|
|
THCTensor *finput,
|
|
THCTensor *fgradInput,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
int64_t inputWidth, int64_t inputHeight,
|
|
int64_t outputWidth, int64_t outputHeight);
|
|
|
|
THC_API void THNN_(SpatialConvolutionLocal_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *weight,
|
|
THCTensor *finput,
|
|
THCTensor *fgradInput,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
int64_t inputWidth, int64_t inputHeight,
|
|
int64_t outputWidth, int64_t outputHeight);
|
|
|
|
THC_API void THNN_(SpatialConvolutionLocal_accGradParameters)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradWeight,
|
|
THCTensor *gradBias,
|
|
THCTensor *finput,
|
|
THCTensor *fgradInput,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
int64_t inputWidth, int64_t inputHeight,
|
|
int64_t outputWidth, int64_t outputHeight,
|
|
accreal scale);
|
|
|
|
THC_API void THNN_(SpatialConvolutionMM_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCTensor *weight,
|
|
THCTensor *bias, // [OPTIONAL]
|
|
THCTensor *columns,
|
|
THCTensor *ones,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH);
|
|
|
|
THC_API void THNN_(SpatialConvolutionMM_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *weight,
|
|
THCTensor *columns,
|
|
THCTensor *ones,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH);
|
|
|
|
THC_API void THNN_(SpatialConvolutionMM_accGradParameters)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradWeight,
|
|
THCTensor *gradBias, // [OPTIONAL]
|
|
THCTensor *columns,
|
|
THCTensor *ones,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
accreal scale);
|
|
|
|
THC_API void THNN_(SpatialDepthwiseConvolution_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCTensor *weight,
|
|
THCTensor *bias, // [OPTIONAL]
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
int dilationW, int dilationH);
|
|
|
|
THC_API void THNN_(SpatialDepthwiseConvolution_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *weight,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
int dilationW, int dilationH);
|
|
|
|
THC_API void THNN_(SpatialDepthwiseConvolution_accGradParameters)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradWeight,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
int dilationW, int dilationH);
|
|
|
|
THC_API void THNN_(SpatialCrossMapLRN_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCTensor *scale,
|
|
int size,
|
|
accreal alpha,
|
|
accreal beta,
|
|
accreal k);
|
|
|
|
THC_API void THNN_(SpatialCrossMapLRN_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *scale,
|
|
THCTensor *output,
|
|
int size,
|
|
accreal alpha,
|
|
accreal beta,
|
|
accreal k);
|
|
|
|
THC_API void THNN_(SpatialDilatedConvolution_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCTensor *weight,
|
|
THCTensor *bias, // [OPTIONAL]
|
|
THCTensor *columns,
|
|
THCTensor *ones,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
int dilationW, int dilationH);
|
|
|
|
THC_API void THNN_(SpatialDilatedConvolution_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *weight,
|
|
THCTensor *columns,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
int dilationW, int dilationH);
|
|
|
|
THC_API void THNN_(SpatialDilatedConvolution_accGradParameters)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradWeight,
|
|
THCTensor *gradBias, // [OPTIONAL]
|
|
THCTensor *columns,
|
|
THCTensor *ones,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
int dilationW, int dilationH,
|
|
accreal scale);
|
|
|
|
THC_API void THNN_(SpatialFullDilatedConvolution_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCTensor *weight,
|
|
THCTensor *bias, // [OPTIONAL]
|
|
THCTensor *columns,
|
|
THCTensor *ones,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
int dilationW, int dilationH,
|
|
int adjW, int adjH);
|
|
|
|
THC_API void THNN_(SpatialFullDilatedConvolution_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *weight,
|
|
THCTensor *columns,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
int dilationW, int dilationH,
|
|
int adjW, int adjH);
|
|
|
|
THC_API void THNN_(SpatialFullDilatedConvolution_accGradParameters)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradWeight,
|
|
THCTensor *gradBias, // [OPTIONAL]
|
|
THCTensor *columns,
|
|
THCTensor *ones,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
int dilationW, int dilationH,
|
|
int adjW, int adjH,
|
|
accreal scale);
|
|
|
|
THC_API void THNN_(SpatialDilatedMaxPooling_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCIndexTensor *indices,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
int dilationW, int dilationH,
|
|
bool ceil_mode);
|
|
|
|
THC_API void THNN_(SpatialDilatedMaxPooling_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCIndexTensor *indices,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
int dilationW, int dilationH,
|
|
bool ceil_mode);
|
|
|
|
THC_API void THNN_(SpatialFullConvolution_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCTensor *weight,
|
|
THCTensor *bias, // [OPTIONAL]
|
|
THCTensor *columns,
|
|
THCTensor *ones,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
int adjW, int adjH);
|
|
|
|
THC_API void THNN_(SpatialFullConvolution_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *weight,
|
|
THCTensor *columns,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
int adjW, int adjH);
|
|
|
|
THC_API void THNN_(SpatialFullConvolution_accGradParameters)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradWeight,
|
|
THCTensor *gradBias, // [OPTIONAL]
|
|
THCTensor *columns,
|
|
THCTensor *ones,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
int adjW, int adjH,
|
|
accreal scale);
|
|
|
|
THC_API void THNN_(SpatialMaxPooling_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCIndexTensor *indices,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
bool ceil_mode);
|
|
|
|
THC_API void THNN_(SpatialMaxPooling_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCIndexTensor *indices,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
int padW, int padH,
|
|
bool ceil_mode);
|
|
|
|
THC_API void THNN_(SpatialMaxUnpooling_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCIndexTensor *indices,
|
|
int owidth, int oheight);
|
|
|
|
THC_API void THNN_(SpatialMaxUnpooling_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCIndexTensor *indices,
|
|
int owidth, int oheight);
|
|
|
|
THC_API void THNN_(SpatialSubSampling_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCTensor *weight,
|
|
THCTensor *bias,
|
|
int kW, int kH,
|
|
int dW, int dH);
|
|
|
|
THC_API void THNN_(SpatialSubSampling_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *weight,
|
|
int kW, int kH,
|
|
int dW, int dH);
|
|
|
|
THC_API void THNN_(SpatialSubSampling_accGradParameters)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradWeight,
|
|
THCTensor *gradBias,
|
|
int kW, int kH,
|
|
int dW, int dH,
|
|
accreal scale);
|
|
|
|
THC_API void THNN_(SpatialUpSamplingBilinear_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
int outputHeight,
|
|
int outputWidth,
|
|
bool align_corners);
|
|
|
|
THC_API void THNN_(SpatialUpSamplingBilinear_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int nbatch,
|
|
int nchannels,
|
|
int inputHeight,
|
|
int inputWidth,
|
|
int outputHeight,
|
|
int outputWidth,
|
|
bool align_corners);
|
|
|
|
THC_API void THNN_(SpatialUpSamplingBicubic_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
int outputHeight,
|
|
int outputWidth,
|
|
bool align_corners);
|
|
|
|
THC_API void THNN_(SpatialUpSamplingBicubic_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int nbatch,
|
|
int nchannels,
|
|
int inputHeight,
|
|
int inputWidth,
|
|
int outputHeight,
|
|
int outputWidth,
|
|
bool align_corners);
|
|
|
|
THC_API void THNN_(SpatialUpSamplingNearest_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int nbatch,
|
|
int nchannels,
|
|
int inputHeight,
|
|
int inputWidth,
|
|
int outputHeight,
|
|
int outputWidth);
|
|
|
|
THC_API void THNN_(SpatialUpSamplingNearest_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
int outputHeight,
|
|
int outputWidth);
|
|
|
|
THC_API void THNN_(RReLU_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCTensor *noise,
|
|
double lower,
|
|
double upper,
|
|
bool train,
|
|
bool inplace,
|
|
void *generator);
|
|
|
|
THC_API void THNN_(RReLU_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *noise,
|
|
double lower,
|
|
double upper,
|
|
bool train,
|
|
bool inplace);
|
|
|
|
THC_API void THNN_(Sigmoid_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output);
|
|
|
|
THC_API void THNN_(Sigmoid_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *output);
|
|
|
|
THC_API void THNN_(SoftMarginCriterion_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *target,
|
|
THCTensor *output,
|
|
int64_t reduction);
|
|
|
|
THC_API void THNN_(SoftMarginCriterion_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *target,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int64_t reduction);
|
|
|
|
THC_API void THNN_(SoftPlus_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
accreal beta,
|
|
accreal threshold);
|
|
|
|
THC_API void THNN_(SoftPlus_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *output,
|
|
accreal beta,
|
|
accreal threshold);
|
|
|
|
THC_API void THNN_(SoftShrink_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
accreal lambda);
|
|
|
|
THC_API void THNN_(SoftShrink_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
accreal lambda);
|
|
|
|
THC_API void THNN_(Square_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output);
|
|
|
|
THC_API void THNN_(Square_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput);
|
|
|
|
THC_API void THNN_(Sqrt_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
accreal eps);
|
|
|
|
THC_API void THNN_(Sqrt_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *output);
|
|
|
|
THC_API void THNN_(Tanh_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output);
|
|
|
|
THC_API void THNN_(Tanh_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *output);
|
|
|
|
THC_API void THNN_(TemporalConvolution_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCTensor *weight,
|
|
THCTensor *bias,
|
|
int kW, int dW,
|
|
int inputFrameSize,
|
|
int outputFrameSize);
|
|
|
|
THC_API void THNN_(TemporalConvolution_updateGradInput)(
|
|
THCState* state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *weight,
|
|
int kW, int dW);
|
|
|
|
THC_API void THNN_(TemporalConvolution_accGradParameters)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradWeight,
|
|
THCTensor *gradBias,
|
|
int kW, int dW,
|
|
accreal scale);
|
|
|
|
THC_API void THNN_(TemporalMaxPooling_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCIndexTensor *indices,
|
|
int kW, int dW);
|
|
|
|
THC_API void THNN_(TemporalMaxPooling_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCIndexTensor *indices,
|
|
int kW, int dW);
|
|
|
|
THC_API void THNN_(TemporalRowConvolution_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCTensor *weight,
|
|
THCTensor *bias, // [OPTIONAL]
|
|
THCTensor *finput,
|
|
THCTensor *fgradInput,
|
|
int kW,
|
|
int dW,
|
|
int padW,
|
|
bool featFirst);
|
|
|
|
THC_API void THNN_(TemporalRowConvolution_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *weight,
|
|
THCTensor *finput,
|
|
THCTensor *fgradInput,
|
|
int kW,
|
|
int dW,
|
|
int padW,
|
|
bool featFirst);
|
|
|
|
THC_API void THNN_(TemporalRowConvolution_accGradParameters)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradWeight,
|
|
THCTensor *gradBias,
|
|
THCTensor *finput,
|
|
THCTensor *fgradInput,
|
|
int kW,
|
|
int dW,
|
|
int padW,
|
|
bool featFirst,
|
|
accreal scale);
|
|
|
|
THC_API void THNN_(TemporalUpSamplingLinear_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
int outputWidth,
|
|
bool align_corners);
|
|
|
|
THC_API void THNN_(TemporalUpSamplingLinear_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int nbatch,
|
|
int nchannels,
|
|
int inputWidth,
|
|
int outputWidth,
|
|
bool align_corners);
|
|
|
|
THC_API void THNN_(TemporalUpSamplingNearest_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int nbatch,
|
|
int nchannels,
|
|
int inputWidth,
|
|
int outputWidth);
|
|
|
|
THC_API void THNN_(TemporalUpSamplingNearest_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
int outputWidth);
|
|
|
|
THC_API void THNN_(VolumetricAveragePooling_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
int kT, int kW, int kH,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH,
|
|
bool ceil_mode,
|
|
bool count_include_pad);
|
|
|
|
THC_API void THNN_(VolumetricAveragePooling_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int kT, int kW, int kH,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH,
|
|
bool ceil_mode,
|
|
bool count_include_pad);
|
|
|
|
// VolumetricConvolution is legacy and purposefully not bound by ATen
|
|
THC_API void THNN_(VolumetricConvolution_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCTensor *weight,
|
|
THCTensor *bias, // [OPTIONAL]
|
|
THCTensor *finput,
|
|
THCTensor *fgradInput,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH);
|
|
|
|
THC_API void THNN_(VolumetricConvolution_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *weight,
|
|
THCTensor *finput,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH);
|
|
|
|
THC_API void THNN_(VolumetricConvolution_accGradParameters)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradWeight,
|
|
THCTensor *gradBias, // [OPTIONAL]
|
|
THCTensor *finput,
|
|
THCTensor *fgradInput,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH,
|
|
accreal scale);
|
|
|
|
THC_API void THNN_(VolumetricDilatedConvolution_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCTensor *weight,
|
|
THCTensor *bias, // [OPTIONAL]
|
|
THCTensor *columns,
|
|
THCTensor *ones,
|
|
int kT, int kW, int kH,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH,
|
|
int dilationT, int dilationW, int dilationH);
|
|
|
|
THC_API void THNN_(VolumetricDilatedConvolution_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *weight,
|
|
THCTensor *columns,
|
|
int kT, int kW, int kH,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH,
|
|
int dilationT, int dilationW, int dilationH);
|
|
|
|
THC_API void THNN_(VolumetricDilatedConvolution_accGradParameters)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradWeight,
|
|
THCTensor *gradBias, // [OPTIONAL]
|
|
THCTensor *columns,
|
|
THCTensor *ones,
|
|
int kT, int kW, int kH,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH,
|
|
int dilationT, int dilationW, int dilationH,
|
|
accreal scale);
|
|
|
|
THC_API void THNN_(VolumetricFullDilatedConvolution_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCTensor *weight,
|
|
THCTensor *bias, // [OPTIONAL]
|
|
THCTensor *finput,
|
|
THCTensor *fgradInput,
|
|
int kT, int kW, int kH,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH,
|
|
int dilationT, int dilationW, int dilationH,
|
|
int adjT, int adjW, int adjH);
|
|
|
|
THC_API void THNN_(VolumetricFullDilatedConvolution_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *weight,
|
|
THCTensor *finput,
|
|
THCTensor *fgradInput,
|
|
int kT, int kW, int kH,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH,
|
|
int dilationT, int dilationW, int dilationH,
|
|
int adjT, int adjW, int adjH);
|
|
|
|
THC_API void THNN_(VolumetricFullDilatedConvolution_accGradParameters)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradWeight, // [OPTIONAL]
|
|
THCTensor *gradBias, // [OPTIONAL]
|
|
THCTensor *finput,
|
|
THCTensor *fgradInput,
|
|
int kT, int kW, int kH,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH,
|
|
int dilationT, int dilationW, int dilationH,
|
|
int adjT, int adjW, int adjH,
|
|
accreal scale);
|
|
|
|
THC_API void THNN_(VolumetricDilatedMaxPooling_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCIndexTensor *indices,
|
|
int kT, int kW, int kH,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH,
|
|
int dilationT, int dilationW, int dilationH,
|
|
bool ceilMode);
|
|
|
|
THC_API void THNN_(VolumetricDilatedMaxPooling_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCIndexTensor *indices,
|
|
int kT, int kW, int kH,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH,
|
|
int dilationT, int dilationW, int dilationH,
|
|
bool ceilMode);
|
|
|
|
THC_API void THNN_(VolumetricFullConvolution_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCTensor *weight,
|
|
THCTensor *bias, // [OPTIONAL]
|
|
THCTensor *finput,
|
|
THCTensor *fgradInput,
|
|
int kT, int kW, int kH,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH,
|
|
int adjT, int adjW, int adjH);
|
|
|
|
THC_API void THNN_(VolumetricFullConvolution_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCTensor *weight,
|
|
THCTensor *finput,
|
|
THCTensor *fgradInput,
|
|
int kT, int kW, int kH,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH,
|
|
int adjT, int adjW, int adjH);
|
|
|
|
THC_API void THNN_(VolumetricFullConvolution_accGradParameters)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradWeight, // [OPTIONAL]
|
|
THCTensor *gradBias, // [OPTIONAL]
|
|
THCTensor *finput,
|
|
THCTensor *fgradInput,
|
|
int kT, int kW, int kH,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH,
|
|
int adjT, int adjW, int adjH,
|
|
accreal scale);
|
|
|
|
THC_API void THNN_(VolumetricMaxPooling_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCIndexTensor *indices,
|
|
int kT, int kW, int kH,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH,
|
|
bool ceilMode);
|
|
|
|
THC_API void THNN_(VolumetricMaxPooling_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCIndexTensor *indices,
|
|
int kT, int kW, int kH,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH,
|
|
bool ceilMode);
|
|
|
|
THC_API void THNN_(VolumetricMaxUnpooling_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCIndexTensor *indices,
|
|
int outputTime, int outputWidth, int outputHeight,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH);
|
|
|
|
THC_API void THNN_(VolumetricMaxUnpooling_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCIndexTensor *indices,
|
|
int outputTime, int outputWidth, int outputHeight,
|
|
int dT, int dW, int dH,
|
|
int padT, int padW, int padH);
|
|
|
|
THC_API void THNN_(VolumetricAdaptiveMaxPooling_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
THCIndexTensor *indices,
|
|
int osizeT,
|
|
int osizeW,
|
|
int osizeH);
|
|
|
|
THC_API void THNN_(VolumetricAdaptiveMaxPooling_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
THCIndexTensor *indices);
|
|
|
|
THC_API void THNN_(VolumetricAdaptiveAveragePooling_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
int osizeT,
|
|
int osizeW,
|
|
int osizeH);
|
|
|
|
THC_API void THNN_(VolumetricAdaptiveAveragePooling_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput);
|
|
|
|
THC_API void THNN_(VolumetricUpSamplingNearest_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int nbatch,
|
|
int nchannels,
|
|
int inputDepth,
|
|
int inputHeight,
|
|
int inputWidth,
|
|
int outputDepth,
|
|
int outputHeight,
|
|
int outputWidth);
|
|
|
|
THC_API void THNN_(VolumetricUpSamplingNearest_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
int outputDepth,
|
|
int outputHeight,
|
|
int outputWidth);
|
|
|
|
THC_API void THNN_(VolumetricUpSamplingTrilinear_updateOutput)(
|
|
THCState *state,
|
|
THCTensor *input,
|
|
THCTensor *output,
|
|
int outputDepth,
|
|
int outputHeight,
|
|
int outputWidth,
|
|
bool align_corners);
|
|
|
|
THC_API void THNN_(VolumetricUpSamplingTrilinear_updateGradInput)(
|
|
THCState *state,
|
|
THCTensor *gradOutput,
|
|
THCTensor *gradInput,
|
|
int nbatch,
|
|
int nchannels,
|
|
int inputDepth,
|
|
int inputHeight,
|
|
int inputWidth,
|
|
int outputDepth,
|
|
int outputHeight,
|
|
int outputWidth,
|
|
bool align_corners);
|
|
|
|
#endif
|