提交 d186ef35 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix crash as we sometimes still have Composite in Composite.

上级 79ba1b7f
...@@ -3658,6 +3658,8 @@ class Composite(ScalarOp): ...@@ -3658,6 +3658,8 @@ class Composite(ScalarOp):
def prepare_node(self, node, storage_map, compute_map): def prepare_node(self, node, storage_map, compute_map):
self.init_py_impls() # self._impls self.init_py_impls() # self._impls
for n in theano.gof.graph.list_of_nodes(self.inputs, self.outputs):
n.op.prepare_node(n, None, None)
def output_types(self, input_types): def output_types(self, input_types):
if tuple(input_types) != self.inputs_type: if tuple(input_types) != self.inputs_type:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论