From 2dd5894283c0cb80e52aa704e06f46f8c0cedf72 Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Mon, 23 Sep 2019 01:00:55 -0400
Subject: [PATCH] Found the names of the input and output tensors, now fixing
 args

Current runtime err:

```
2019-09-23 01:00:01.189765: W
tensorflow/core/framework/op_kernel.cc:1502] OP_REQUIRES failed at
transpose_op.cc:157 : Invalid argument: transpose expects a vector of
size 1. But input(1) is a vector of size 3
thread 'main' panicked at 'Run success: {inner:0x5589ea8b3860,
InvalidArgument: transpose expects a vector of size 1. But input(1) is a
vector of size 3
```

Getting there...!
---
 src/main.rs | 28 ++++++++++++++++++++++------
 1 file changed, 22 insertions(+), 6 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index aa28823a8e..a1ffd511f1 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -22,12 +22,15 @@ use tensorflow::Status;
 use tensorflow::Tensor;
 use tensorflow::version;
 
+const INPUT_NAME : &'static str = "input_A_test";
+const OUTPUT_NAME : &'static str = "generator_A2B_3/output_transpose";
+
 fn main() {
   print_version();
   load_model();
   //load_model_2();
   run_audio().expect("should work");
-  run_audio().expect("should work");
+  //run_audio().expect("should work");
 }
 
 fn print_version() {
@@ -52,19 +55,32 @@ fn load_model() {
   //graph.import_graph_def(&proto, &ImportGraphDefOptions::new())?;
   //let session = Session::new(&SessionOptions::new(), &graph)?;
 
-  // Run the graph.
   let mut x = Tensor::new(&[1]);
-  x[0] = 2i32;
+  x[0] = 2.0f32;
+
   let mut y = Tensor::new(&[1]);
   y[0] = 40i32;
 
+  // Run the graph.
   let mut args = SessionRunArgs::new();
-  args.add_feed(&graph.operation_by_name_required("input_A_test").expect("x"), 0, &x);
-  //args.add_feed(&graph.operation_by_name_required("y").expect("y"), 0, &y);
-  //let z = args.request_fetch(&graph.operation_by_name_required("z").expect("z"), 0);
+  
+  // input_A_test:
+  // Tensor("input_A_test:0", shape=(?, 24, ?), dtype=float32)
+  args.add_feed(&graph.operation_by_name_required(INPUT_NAME)
+      .expect(INPUT_NAME), 0, &x);
+
+  // generation_B_test:
+  // Tensor("generator_A2B_3/output_transpose:0", shape=(?, 24, ?), dtype=float32)
+  let _z = args.request_fetch(
+    &graph.operation_by_name_required(OUTPUT_NAME)
+        .expect(OUTPUT_NAME), 0);
 
   session.run(&mut args).expect("Run success");
 
+  // Check our results.
+  //let z_res: i32 = args.fetch(z).expect("ret")[0];
+  //println!("{:?}", z_res);
+
 }
 
 fn load_model_2() {