diff options
-rw-r--r-- | gmodule.lua | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/gmodule.lua b/gmodule.lua index 551baae..0a81da9 100644 --- a/gmodule.lua +++ b/gmodule.lua @@ -265,6 +265,12 @@ function gModule:type(type, tensorCache) end end + for i,node in ipairs(self.forwardnodes) do + if node.data.input ~= nil then + node.data.input = recursiveType(node.data.input, type) + end + end + self._type = type return self end |