diff options
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 |