diff options
| author | Adam Lerer <alerer@fb.com> | 2015-06-25 00:44:29 +0300 |
|---|---|---|
| committer | Adam Lerer <alerer@fb.com> | 2015-09-04 10:00:01 +0300 |
| commit | dd9d7fd996a476ef0af0e39b7039a761745df943 (patch) | |
| tree | ec7586422a8baf12e9e3c28bdd34229e1f444651 /Euclidean.lua | |
| parent | bab75d48973ce1767a575fa971a2fb9dc7363ff4 (diff) | |
nn.Module preserve type sharing semantics (#187); add nn.Module.apply
Diffstat (limited to 'Euclidean.lua')
| -rw-r--r-- | Euclidean.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/Euclidean.lua b/Euclidean.lua index ae3fee9..25aa2e4 100644 --- a/Euclidean.lua +++ b/Euclidean.lua @@ -171,7 +171,7 @@ function Euclidean:accGradParameters(input, gradOutput, scale) end end -function Euclidean:type(type) +function Euclidean:type(type, tensorCache) if type then -- prevent premature memory allocations self._input = nil @@ -186,5 +186,5 @@ function Euclidean:type(type) self._repeat = nil self._repeat2 = nil end - return parent.type(self, type) + return parent.type(self, type, tensorCache) end |