diff --git a/src/TensorFlowNET.Core/Eager/EagerRunner.TFE_FastPathExecute.cs b/src/TensorFlowNET.Core/Eager/EagerRunner.TFE_FastPathExecute.cs index 6bca3319..0aef39a9 100644 --- a/src/TensorFlowNET.Core/Eager/EagerRunner.TFE_FastPathExecute.cs +++ b/src/TensorFlowNET.Core/Eager/EagerRunner.TFE_FastPathExecute.cs @@ -50,6 +50,9 @@ namespace Tensorflow.Eager var op_def = tf.get_default_graph().GetOpDef(opName); + var flattened_attrs = new List(op_def.InputArg.Count); + var flattened_inputs = new List(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 @@ namespace Tensorflow.Eager 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(op_def.InputArg.Count); - var flattened_inputs = new List(op_def.InputArg.Count); - c_api.TFE_OpSetDevice(op, device_name, status.Handle); status.Check(true); @@ -229,6 +232,7 @@ namespace Tensorflow.Eager default: var tensor = tf.convert_to_tensor(inputs); input_handle = tensor.EagerTensorHandle; + flattened_inputs.Add(tensor); break; }