提交 a8955e3d authored 作者: ChienliMa's avatar ChienliMa

Modify link.map_storage() so that it make use of given storage_map

上级 f6c74fcc
...@@ -547,7 +547,6 @@ def map_storage(fgraph, order, input_storage, output_storage, storage_map=None): ...@@ -547,7 +547,6 @@ def map_storage(fgraph, order, input_storage, output_storage, storage_map=None):
if storage_map is None: if storage_map is None:
storage_map = {} storage_map = {}
# add input storage into storage_map # add input storage into storage_map
for r, storage in zip(fgraph.inputs, input_storage): for r, storage in zip(fgraph.inputs, input_storage):
storage_map[r] = storage storage_map[r] = storage
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论