Asserts that two GraphDefs are (mostly) the same.
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.
|
TensorFlow 2 version
View source on GitHub