Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/torch/cunn.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'lib/THCUNN/generic/SparseLinear.cu')
-rw-r--r--lib/THCUNN/generic/SparseLinear.cu10
1 files changed, 5 insertions, 5 deletions
diff --git a/lib/THCUNN/generic/SparseLinear.cu b/lib/THCUNN/generic/SparseLinear.cu
index 6838cac..f22b233 100644
--- a/lib/THCUNN/generic/SparseLinear.cu
+++ b/lib/THCUNN/generic/SparseLinear.cu
@@ -127,8 +127,8 @@ void THNN_(SparseLinear_accGradParameters)(
THCTensor *gradBias,
THCTensor *weight,
THCTensor *bias,
- accreal weightDecay,
- accreal scale)
+ double weightDecay,
+ double scale)
{
long outDim = THCTensor_(size)(state, weight, 0);
long inDim = THCTensor_(size)(state, weight, 1);
@@ -237,8 +237,8 @@ void THNN_(SparseLinear_legacyAccGradParameters)(
THCTensor *gradBias,
THCTensor *weight,
THCTensor *bias,
- accreal weightDecay,
- accreal scale) {
+ double weightDecay,
+ double scale) {
THError("CUDA does not support legacy input format, please use a table of nnz x 2 vectors");
}
@@ -259,7 +259,7 @@ void THNN_(SparseLinear_updateParameters)(
THCTensor *gradWeight,
THCTensor *gradBias,
THCTensor *lastInput,
- accreal learningRate) {
+ double learningRate) {
THCTensor_(cadd)(state, weight, weight, -learningRate, gradWeight);
THCTensor_(cadd)(state, bias, bias, -learningRate, gradBias);
}