rename target operation name

Author: dskkato

examples/addition.rs

@@ -12,10 +12,6 @@ use tensorflow::SessionRunArgs;
 use tensorflow::Status;
 use tensorflow::Tensor;
 
-#[cfg_attr(feature = "examples_system_alloc", global_allocator)]
-#[cfg(feature = "examples_system_alloc")]
-static ALLOCATOR: std::alloc::System = std::alloc::System;
-
 fn main() -> Result<(), Box<dyn Error>> {
     let filename = "examples/addition/model.pb"; // z = x + y
     if !Path::new(filename).exists() {
@@ -49,7 +45,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

@@ -7,19 +7,16 @@
 
 @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}")
-
 logdir = "logs/addition"
 writer = tf.summary.create_file_writer(logdir)
 with writer.as_default():