#ifndef THC_GENERIC_FILE #define THC_GENERIC_FILE "THCUNN/generic/THCUNN.h" #else #include 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_(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_(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_(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_(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); #endif