rename add operation to z

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("Identity")?, 0);
+    let z = args.request_fetch(&graph.operation_by_name_required("z")?, 0);
     session.run(&mut args)?;
 
     // Check our results.

examples/addition/addition.py

@@ -6,15 +6,11 @@
 
 @tf.function
 def add(x, y):
-    return tf.add(x, y)
+    tf.add(x, y, name='z')
 
 x = tf.TensorSpec((), dtype=tf.dtypes.int32, name='x')
 y = tf.TensorSpec((), dtype=tf.dtypes.int32, name='y')
 
 concrete_function = add.get_concrete_function(x, y)
 directory = 'examples/addition'
 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}')