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 /MixtureTable.lua | |
| parent | bab75d48973ce1767a575fa971a2fb9dc7363ff4 (diff) | |
nn.Module preserve type sharing semantics (#187); add nn.Module.apply
Diffstat (limited to 'MixtureTable.lua')
| -rw-r--r-- | MixtureTable.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/MixtureTable.lua b/MixtureTable.lua index 16e5832..16dafc4 100644 --- a/MixtureTable.lua +++ b/MixtureTable.lua @@ -149,7 +149,7 @@ function MixtureTable:updateGradInput(input, gradOutput) return self.gradInput end -function MixtureTable:type(type) +function MixtureTable:type(type, tensorCache) self._gaterView = nil self._expert = nil self._expertView = nil @@ -157,5 +157,5 @@ function MixtureTable:type(type) self._gradInput = nil self._expert2 = nil self._expertView2 = nil - return parent.type(self, type) + return parent.type(self, type, tensorCache) end |