diff options
author | Adam Paszke <adam.paszke@gmail.com> | 2016-12-28 02:36:07 +0300 |
---|---|---|
committer | Adam Paszke <adam.paszke@gmail.com> | 2016-12-28 02:36:11 +0300 |
commit | 0826d635be8b9825aaf7dff96d6a3847708865fa (patch) | |
tree | b11b875186831c26808171702693526750977713 | |
parent | d3d21f5058b323c9b291d5998e0ea593fea4dec3 (diff) |
Use bool for sizeAverage in SoftMarginCriterion
-rw-r--r-- | lib/THCUNN/generic/SoftMarginCriterion.cu | 4 | ||||
-rw-r--r-- | lib/THCUNN/generic/THCUNN.h | 4 |
2 files changed, 4 insertions, 4 deletions
diff --git a/lib/THCUNN/generic/SoftMarginCriterion.cu b/lib/THCUNN/generic/SoftMarginCriterion.cu index 1ae9307..304f8b3 100644 --- a/lib/THCUNN/generic/SoftMarginCriterion.cu +++ b/lib/THCUNN/generic/SoftMarginCriterion.cu @@ -7,7 +7,7 @@ void THNN_(SoftMarginCriterion_updateOutput)( THCTensor *input, THCTensor *target, THCTensor *output, - int sizeAverage) + bool sizeAverage) { THCUNN_check_nElement(state, input, target); THCUNN_check_dim_size(state, output, 1, 0, 1); @@ -37,7 +37,7 @@ void THNN_(SoftMarginCriterion_updateGradInput)( THCTensor *input, THCTensor *target, THCTensor *gradInput, - int sizeAverage) + bool sizeAverage) { THCUNN_check_nElement(state, input, target); THCUNN_assertSameGPU(state, 3, input, target, gradInput); diff --git a/lib/THCUNN/generic/THCUNN.h b/lib/THCUNN/generic/THCUNN.h index d5b7996..bf903b9 100644 --- a/lib/THCUNN/generic/THCUNN.h +++ b/lib/THCUNN/generic/THCUNN.h @@ -805,14 +805,14 @@ TH_API void THNN_(SoftMarginCriterion_updateOutput)( THCTensor *input, THCTensor *target, THCTensor *output, - int sizeAverage); + bool sizeAverage); TH_API void THNN_(SoftMarginCriterion_updateGradInput)( THCState *state, THCTensor *input, THCTensor *target, THCTensor *gradInput, - int sizeAverage); + bool sizeAverage); TH_API void THNN_(SoftMax_updateOutput)( THCState *state, |