fix attrs missing.

Oceania2018 · Oceania2018 · commit 601c42341cd0 · 2020-07-25T19:08:38.000-05:00
diff --git a/src/TensorFlowNET.Core/Eager/EagerRunner.TFE_FastPathExecute.cs b/src/TensorFlowNET.Core/Eager/EagerRunner.TFE_FastPathExecute.cs
@@ -50,6 +50,9 @@ public Tensor[] TFE_FastPathExecute(Context ctx,
 
             var op_def = tf.get_default_graph().GetOpDef(opName);
 
+            var flattened_attrs = new List<object>(op_def.InputArg.Count);
+            var flattened_inputs = new List<Tensor>(op_def.InputArg.Count);
+
             // Set non-inferred attrs, including setting defaults if the attr is passed in
             // as None.
             for (int i = kFastPathExecuteInputStartIndex + op_def.InputArg.Count; i < args_size; i += 2)
@@ -60,14 +63,14 @@ public Tensor[] TFE_FastPathExecute(Context ctx,
                 var attr = op_def.Attr.FirstOrDefault(x => x.Name == attr_name);
                 if(attr != null)
                 {
+                    flattened_attrs.Add(attr_name);
+                    flattened_attrs.Add(attr_value);
+
                     SetOpAttrWithDefaults(ctx, op, attr, attr_name, attr_value, attr_list_sizes, status);
                     status.Check(true);
                 }
             }
 
-            var flattened_attrs = new List<object>(op_def.InputArg.Count);
-            var flattened_inputs = new List<Tensor>(op_def.InputArg.Count);
-
             c_api.TFE_OpSetDevice(op, device_name, status.Handle);
             status.Check(true);
 
@@ -229,6 +232,7 @@ bool AddInputToOp(object inputs,
                 default:
                     var tensor = tf.convert_to_tensor(inputs);
                     input_handle = tensor.EagerTensorHandle;
+                    flattened_inputs.Add(tensor);
                     break;
             }
 
