提交 55277c43 authored 作者: ChienliMa's avatar ChienliMa

Add some docs to FunctionGraph and minor changes of function_module.copy()

上级 48461734
......@@ -255,17 +255,26 @@ class T_function(unittest.TestCase):
fgraph_ori = ori.maker.fgraph
fgraph_cpy = cpy.maker.fgraph
<<<<<<< HEAD
# assert intermediate and Constants storages are shared.
# and output stoarges are not shared
=======
# assert intermediate and Constants storages are shared
>>>>>>> Add some docs to FunctionGraph and minor changes of function_module.copy()
i_o_variables = fgraph_cpy.inputs + fgraph_cpy.outputs
ori_storages = storage_map_ori.values()
for key in storage_map_cpy.keys():
storage = storage_map_cpy[key]
storage_is_shared = any([ storage is s for s in ori_storages])
if key not in i_o_variables or isinstance(key, theano.tensor.Constant):
<<<<<<< HEAD
self.assertTrue(storage_is_shared)
elif key in fgraph_cpy.outputs:
self.assertFalse(storage_is_shared)
=======
storage = storage_map_cpy[key]
self.assertTrue( any([ storage is s for s in ori_storages]))
>>>>>>> Add some docs to FunctionGraph and minor changes of function_module.copy()
# assert storages of SharedVariable without updates are shared
for (input, _1, _2), here, there in zip(ori.indices,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论