update addition example's python code for tf2.5

Author: dskkato

examples/addition.rs

@@ -49,7 +49,7 @@ fn main() -> Result<(), Box<dyn Error>> {
     let mut args = SessionRunArgs::new();
     args.add_feed(&graph.operation_by_name_required("x")?, 0, &x);
     args.add_feed(&graph.operation_by_name_required("y")?, 0, &y);
-    let z = args.request_fetch(&graph.operation_by_name_required("z")?, 0);
+    let z = args.request_fetch(&graph.operation_by_name_required("Identity")?, 0);
     session.run(&mut args)?;
 
     // Check our results.

examples/addition/addition.py

@@ -1,14 +1,20 @@
-# TODO: Stop using v1 compatibility
-import tensorflow.compat.v1 as tf
+import tensorflow as tf
 
+# check tensorflow version is 2.x 
+tf_major_version = tf.__version__.split('.')[0]
+assert tf_major_version == '2'
 
-tf.disable_eager_execution()
-x = tf.placeholder(tf.int32, name = 'x')
-y = tf.placeholder(tf.int32, name = 'y')
-z = tf.add(x, y, name = 'z')
+@tf.function
+def add(x, y):
+    return tf.add(x, y)
 
-tf.variables_initializer(tf.global_variables(), name = 'init')
+x = tf.TensorSpec((), dtype=tf.dtypes.int32, name='x')
+y = tf.TensorSpec((), dtype=tf.dtypes.int32, name='y')
 
-definition = tf.Session().graph_def
+concrete_function = add.get_concrete_function(x, y)
 directory = 'examples/addition'
-tf.train.write_graph(definition, directory, 'model.pb', as_text=False)
+tf.io.write_graph(concrete_function.graph, directory, 'model.pb', as_text=False)
+
+# check inputs/outputs node names to refer from Rust later on
+print(f'input nodes  : {concrete_function.inputs}')
+print(f'output nodes : {concrete_function.outputs}')