From 601c42341cd06bc5f54f168cfa5a0b725686f2fb Mon Sep 17 00:00:00 2001 From: Oceania2018 Date: Sat, 25 Jul 2020 19:08:38 -0500 Subject: [PATCH] fix attrs missing. --- .../Eager/EagerRunner.TFE_FastPathExecute.cs | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) 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; }