Update cmsml tool usage in existing TensorFlow tests.

cms-sw · Feb 15, 2024 · 9784e09 · 9784e09
1 parent e08b8ba
commit 9784e09
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/PhysicsTools/TensorFlow/test/createconstantgraph.py b/PhysicsTools/TensorFlow/test/createconstantgraph.py
@@ -47,4 +47,4 @@
 # write it
 graph_path = os.path.join(datadir, "constantgraph.pb")
 outputs = ["output"]
-cmsml.tensorflow.save_graph(graph_path, sess, output_names=outputs, variables_to_constants=True)
+cmsml.tensorflow.save_frozen_graph(graph_path, sess, output_names=outputs, variables_to_constants=True)
diff --git a/PhysicsTools/TensorFlow/test/readconstantgraph.py b/PhysicsTools/TensorFlow/test/readconstantgraph.py
@@ -24,5 +24,5 @@
 
 # read and evaluate the graph
 graph_path = os.path.join(datadir, "constantgraph.pb")
-graph, sess = cmsml.tensorflow.load_graph(graph_path, create_session=True)
+graph, sess = cmsml.tensorflow.load_frozen_graph(graph_path, create_session=True)
 print(sess.run("output:0", feed_dict={"scale:0": 1.0, "input:0": [range(10)]})[0][0])