diff options
Diffstat (limited to 'tensorflow/core/framework/graph_def_util.cc')
-rw-r--r-- | tensorflow/core/framework/graph_def_util.cc | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/tensorflow/core/framework/graph_def_util.cc b/tensorflow/core/framework/graph_def_util.cc new file mode 100644 index 0000000000..1e0d280126 --- /dev/null +++ b/tensorflow/core/framework/graph_def_util.cc @@ -0,0 +1,25 @@ +#include "tensorflow/core/framework/graph_def_util.h" + +#include "tensorflow/core/framework/node_def_util.h" +#include "tensorflow/core/lib/core/errors.h" +#include "tensorflow/core/lib/strings/strcat.h" +#include "tensorflow/core/public/status.h" + +namespace tensorflow { + +string SummarizeGraphDef(const GraphDef& graph_def) { + string ret; + for (const NodeDef& node : graph_def.node()) { + strings::StrAppend(&ret, SummarizeNodeDef(node), ";\n"); + } + return ret; +} + +Status ValidateExternalGraphDefSyntax(const GraphDef& graph_def) { + for (const NodeDef& node : graph_def.node()) { + TF_RETURN_IF_ERROR(ValidateExternalNodeDefSyntax(node)); + } + return Status::OK(); +} + +} // namespace tensorflow |