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

github.com/torch/nn.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorSoumith Chintala <soumith@gmail.com>2017-08-26 21:47:55 +0300
committerSoumith Chintala <soumith@gmail.com>2017-08-26 21:47:55 +0300
commitc5972818f19fbc40855137d86d1b175c8dc04a5a (patch)
tree3678609d52b0bff54e9acb465bc16f1ed75a95e9
parent6cfd1dd69bc95813439596724f60513e269d6537 (diff)
fix VolumetricAveragePooling for changed C interface
-rw-r--r--VolumetricAveragePooling.lua8
1 files changed, 6 insertions, 2 deletions
diff --git a/VolumetricAveragePooling.lua b/VolumetricAveragePooling.lua
index df6d2c4..146304a 100644
--- a/VolumetricAveragePooling.lua
+++ b/VolumetricAveragePooling.lua
@@ -21,7 +21,9 @@ function VolumetricAveragePooling:updateOutput(input)
input:cdata(),
self.output:cdata(),
self.kT, self.kW, self.kH,
- self.dT, self.dW, self.dH
+ self.dT, self.dW, self.dH,
+ 0, 0, 0,
+ false, true
)
return self.output
end
@@ -32,7 +34,9 @@ function VolumetricAveragePooling:updateGradInput(input, gradOutput)
gradOutput:cdata(),
self.gradInput:cdata(),
self.kT, self.kW, self.kH,
- self.dT, self.dW, self.dH
+ self.dT, self.dW, self.dH,
+ 0, 0, 0,
+ false, true
)
return self.gradInput
end