tf.test.assert_equal_graph_def

View source on GitHub

Asserts that two GraphDefs are (mostly) the same.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.test.assert_equal_graph_def

tf.test.assert_equal_graph_def(
    actual, expected, checkpoint_v2=False, hash_table_shared_name=False
)

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.

Args
actual	The GraphDef we have.
expected	The GraphDef we expected.
checkpoint_v2	boolean determining whether to ignore randomized attribute values that appear in V2 checkpoints.
hash_table_shared_name	boolean determining whether to ignore randomized shared_names that appear in HashTableV2 op defs.

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