From f41959ccb2d9d4c722fe8fc3351401d53bcf4900 Mon Sep 17 00:00:00 2001 From: Manjunath Kudlur Date: Fri, 6 Nov 2015 16:27:58 -0800 Subject: TensorFlow: Initial commit of TensorFlow library. TensorFlow is an open source software library for numerical computation using data flow graphs. Base CL: 107276108 --- tensorflow/core/graph/equal_graph_def.h | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 tensorflow/core/graph/equal_graph_def.h (limited to 'tensorflow/core/graph/equal_graph_def.h') diff --git a/tensorflow/core/graph/equal_graph_def.h b/tensorflow/core/graph/equal_graph_def.h new file mode 100644 index 0000000000..7dd8aab340 --- /dev/null +++ b/tensorflow/core/graph/equal_graph_def.h @@ -0,0 +1,32 @@ +#ifndef TENSORFLOW_GRAPH_EQUAL_GRAPH_DEF_H_ +#define TENSORFLOW_GRAPH_EQUAL_GRAPH_DEF_H_ + +#include "tensorflow/core/framework/graph.pb.h" +#include "tensorflow/core/framework/graph_def_util.h" +#include "tensorflow/core/platform/port.h" + +namespace tensorflow { + +// Determines if actual and expected are equal, ignoring ordering of +// nodes, attrs, and control inputs. If the GraphDefs are different +// and diff != nullptr, *diff is set to an explanation of the +// difference. Note that we use node names to match up nodes between +// the graphs, and so the naming of nodes must be consistent. +bool EqualGraphDef(const GraphDef& actual, const GraphDef& expected, + string* diff); + +// Determines if actual and expected are equal, ignoring ordering of +// attrs and control inputs. If the NodeDefs are different and +// diff != nullptr, *diff is set to an explanation of the difference. +bool EqualNodeDef(const NodeDef& actual, const NodeDef& expected, string* diff); + +#define TF_EXPECT_GRAPH_EQ(expected, actual) \ + do { \ + string diff; \ + EXPECT_TRUE(EqualGraphDef(actual, expected, &diff)) \ + << diff << "\nActual: " << SummarizeGraphDef(actual); \ + } while (false) + +} // namespace tensorflow + +#endif // TENSORFLOW_GRAPH_EQUAL_GRAPH_DEF_H_ -- cgit v1.2.3