OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Mark optional arguments in THNN.h

Browse Source
This commit is contained in:
Adam Paszke 2016-08-04 08:43:13 -07:00
parent d920bf1488
commit 4fe7059a31
1 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
generic/THNN.h
View File

@ -124,7 +124,7 @@ TH_API void THNN_(L1Cost_updateOutput)(
TH_API void THNN_(L1Cost_updateGradInput)( TH_API void THNN_(L1Cost_updateGradInput)(
THNNState *state, // library's state THNNState *state, // library's state
THTensor *input, // input tensor THTensor *input, // input tensor
THTensor *gradOutput, // gradient w.r.t module's output THTensor *gradOutput, // [OPTIONAL] gradient w.r.t module's output
THTensor *gradInput); // [OUT] gradient w.r.t the input THTensor *gradInput); // [OUT] gradient w.r.t the input
TH_API void THNN_(LeakyReLU_updateOutput)( TH_API void THNN_(LeakyReLU_updateOutput)(
@ -170,8 +170,8 @@ TH_API void THNN_(LookupTable_accGradParameters)(
THTensor *gradOutput, THTensor *gradOutput,
THTensor *gradWeight, THTensor *gradWeight,
THIntegerTensor *count, THIntegerTensor *count,
THTensor *sorted, THTensor *sorted, // [OPTIONAL]
THTensor *indices, THTensor *indices, // [OPTIONAL]
bool scaleGradByFreq, bool scaleGradByFreq,
int paddingValue, int paddingValue,
real scale); real scale);
@ -247,7 +247,7 @@ TH_API void THNN_(MultiMarginCriterion_updateOutput)(
THTensor *output, THTensor *output,
bool sizeAverage, bool sizeAverage,
int p, int p,
THTensor* weights, THTensor* weights, // [OPTIONAL]
real margin); real margin);
TH_API void THNN_(MultiMarginCriterion_updateGradInput)( TH_API void THNN_(MultiMarginCriterion_updateGradInput)(
THNNState *state, THNNState *state,
@ -256,7 +256,7 @@ TH_API void THNN_(MultiMarginCriterion_updateGradInput)(
THTensor *gradInput, THTensor *gradInput,
bool sizeAverage, bool sizeAverage,
int p, int p,
THTensor *weights, THTensor *weights, // [OPTIONAL]
real margin); real margin);
TH_API void THNN_(PReLU_updateOutput)( TH_API void THNN_(PReLU_updateOutput)(
@ -539,8 +539,8 @@ TH_API void THNN_(BatchNormalization_updateOutput)(
THNNState *state, THNNState *state,
THTensor *input, THTensor *input,
THTensor *output, THTensor *output,
THTensor *weight, THTensor *weight, // [OPTIONAL]
THTensor *bias, THTensor *bias, // [OPTIONAL]
THTensor *running_mean, THTensor *running_mean,
THTensor *running_var, THTensor *running_var,
THTensor *save_mean, THTensor *save_mean,
@ -552,10 +552,10 @@ TH_API void THNN_(BatchNormalization_backward)(
THNNState *state, THNNState *state,
THTensor *input, THTensor *input,
THTensor *gradOutput, THTensor *gradOutput,
THTensor *gradInput, THTensor *gradInput, // [OPTIONAL]
THTensor *gradWeight, THTensor *gradWeight, // [OPTIONAL]
THTensor *gradBias, THTensor *gradBias, // [OPTIONAL]
THTensor *weight, THTensor *weight, // [OPTIONAL]
THTensor *running_mean, THTensor *running_mean,
THTensor *running_var, THTensor *running_var,
THTensor *save_mean, THTensor *save_mean,
@ -602,7 +602,7 @@ TH_API void THNN_(SpatialConvolutionMM_updateOutput)(
THTensor *input, THTensor *input,
THTensor *output, THTensor *output,
THTensor *weight, THTensor *weight,
THTensor *bias, THTensor *bias, // [OPTIONAL]
THTensor *finput, THTensor *finput,
THTensor *fgradInput, THTensor *fgradInput,
int kW, int kH, int kW, int kH,
@ -624,7 +624,7 @@ TH_API void THNN_(SpatialConvolutionMM_accGradParameters)(
THTensor *input, THTensor *input,
THTensor *gradOutput, THTensor *gradOutput,
THTensor *gradWeight, THTensor *gradWeight,
THTensor *gradBias, THTensor *gradBias, // [OPTIONAL]
THTensor *finput, THTensor *finput,
THTensor *fgradInput, THTensor *fgradInput,
int kW, int kH, int kW, int kH,
@ -728,7 +728,7 @@ TH_API void THNN_(SpatialFullConvolution_updateOutput)(
THTensor *input, THTensor *input,
THTensor *output, THTensor *output,
THTensor *weight, THTensor *weight,
THTensor *bias, THTensor *bias, // [OPTIONAL]
THTensor *columns, THTensor *columns,
THTensor *ones, THTensor *ones,
int kW, int kH, int kW, int kH,
@ -751,7 +751,7 @@ TH_API void THNN_(SpatialFullConvolution_accGradParameters)(
THTensor *input, THTensor *input,
THTensor *gradOutput, THTensor *gradOutput,
THTensor *gradWeight, THTensor *gradWeight,
THTensor *gradBias, THTensor *gradBias, // [OPTIONAL]
THTensor *columns, THTensor *columns,
THTensor *ones, THTensor *ones,
int kW, int kH, int kW, int kH,
@ -798,7 +798,7 @@ TH_API void THNN_(SpatialDilatedConvolution_updateOutput)(
THTensor *input, THTensor *input,
THTensor *output, THTensor *output,
THTensor *weight, THTensor *weight,
THTensor *bias, THTensor *bias, // [OPTIONAL]
THTensor *columns, THTensor *columns,
THTensor *ones, THTensor *ones,
int kW, int kH, int kW, int kH,
@ -823,7 +823,7 @@ TH_API void THNN_(SpatialDilatedConvolution_accGradParameters)(
THTensor *input, THTensor *input,
THTensor *gradOutput, THTensor *gradOutput,
THTensor *gradWeight, THTensor *gradWeight,
THTensor *gradBias, THTensor *gradBias, // [OPTIONAL]
THTensor *columns, THTensor *columns,
THTensor *ones, THTensor *ones,
int kW, int kH, int kW, int kH,