tf.test.assert_equal_graph_def

View source on GitHub

Asserts that two GraphDefs are (mostly) the same.

tf.test.assert_equal_graph_def(
    expected, actual
)

Compares two GraphDef protos for equality, ignoring versions and ordering of nodes, attrs, and control inputs. Node names are used to match up nodes between the graphs, so the naming of nodes must be consistent. This function ignores randomized attribute values that may appear in V2 checkpoints.

Args
expected	The GraphDef we expected.
actual	The GraphDef we have.

Raises
AssertionError	If the GraphDefs do not match.
TypeError	If either argument is not a GraphDef.