tf.test.assert_equal_graph_def  |  TensorFlow v2.16.1 (original) (raw)

tf.test.assert_equal_graph_def

Stay organized with collections Save and categorize content based on your preferences.

Asserts that two GraphDefs are (mostly) the same.

tf.test.assert_equal_graph_def(
    expected: tf.compat.v1.GraphDef,
    actual: tf.compat.v1.GraphDef
) -> None

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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-04-26 UTC.