search

CODAIT / graph_def_editor

GraphDef Editor: A port of the TensorFlow contrib.graph_editor package that operates over serialized graphs
Apache License 2.0
31 stars 16 forks source link

Graph copying doesn't handle collections properly #36

Closed frreiss closed 5 years ago

frreiss commented 5 years ago

If you call gde.copy(g1, g2) and graph g1 contains collections, then GDE attempts to call the add_to_collection() TensorFlow API on g2, as if g2 were a tf.Graph instead of a gde.Graph.

The function assign_renamed_collections_handler() in transform.py needs to be rewritten to use GDE APIs to manage collections in the target graph.