AddControlInput and UpdateEdge works..

Author: deepakkumar1984

src/TensorFlowNET.Core/Graphs/python_api.graph.cs

@@ -1,52 +1,52 @@
-using System;
-using System.Collections.Generic;
-using System.Text;
-using Tensorflow.Operations;
-
-namespace Tensorflow
-{
-    public partial class Operation
-    {
+using System;
+using System.Collections.Generic;
+using System.Text;
+using Tensorflow.Operations;
+
+namespace Tensorflow
+{
+    public partial class Operation
+    {
         private ControlFlowContext _control_flow_context;
 
-        /// <summary>
-        /// Add this op to its control flow context.
-        /// 
-        /// This may add new ops and change this op's inputs. self.inputs must be
-        /// available before calling this method.
-        /// </summary>
-        public void _control_flow_post_processing()
-        {
-            foreach(var input_tensor in inputs)
-            {
-                //TODO: implement below code dependency
-                //control_flow_util.CheckInputFromValidContext(this, input_tensor.op);
-            }
-
-            if (_control_flow_context != null)
-                _control_flow_context.AddOp(this);
-        }
-
-        public void _add_control_input(Operation op)
-        {
-            //c_api.TF_AddControlInput(_operDesc, op);
-            c_api.AddControlInput(graph, _handle, op);
-        }
-
-        public void _add_control_inputs(Operation[] ops)
-        {
-            foreach (var op in ops)
-                _add_control_input(op);
-        }
-
-        public void _set_control_flow_context(ControlFlowContext ctx)
-        {
-            _control_flow_context = ctx;
-        }
-
-        public ControlFlowContext _get_control_flow_context()
-        {
-            return _control_flow_context;
-        }
-    }
-}
+        /// <summary>
+        /// Add this op to its control flow context.
+        /// 
+        /// This may add new ops and change this op's inputs. self.inputs must be
+        /// available before calling this method.
+        /// </summary>
+        public void _control_flow_post_processing()
+        {
+            foreach(var input_tensor in inputs)
+            {
+                //TODO: implement below code dependency
+                //control_flow_util.CheckInputFromValidContext(this, input_tensor.op);
+            }
+
+            if (_control_flow_context != null)
+                _control_flow_context.AddOp(this);
+        }
+
+        public void _add_control_input(Operation op)
+        {
+            //c_api.TF_AddControlInput(_operDesc, op);
+            c_api.AddControlInput(graph, _handle, op);
+        }
+
+        public void _add_control_inputs(Operation[] ops)
+        {
+            foreach (var op in ops)
+                _add_control_input(op);
+        }
+
+        public void _set_control_flow_context(ControlFlowContext ctx)
+        {
+            _control_flow_context = ctx;
+        }
+
+        public ControlFlowContext _get_control_flow_context()
+        {
+            return _control_flow_context;
+        }
+    }
+}

src/TensorFlowNET.Core/Operations/Operation.Control.cs

@@ -30,6 +30,8 @@ public static Tensor zeros(Shape shape, TF_DataType dtype = TF_DataType.TF_FLOAT
                         return _constant_if_small(0.0F, shape, dtype, name);
                     case TF_DataType.TF_INT32:
                         return _constant_if_small(0, shape, dtype, name);
+                    case TF_DataType.TF_INT8:
+                        return _constant_if_small<byte>(0, shape, dtype, name);
                     default:
                         throw new TypeError("can't find type for zeros");
                 }

src/TensorFlowNET.Core/Operations/array_ops.py.cs

@@ -129,10 +129,8 @@ public static Tensor zero_fraction(Tensor value, string name = null)
         {
             return with(ops.name_scope(name, "zero_fraction", new { value }), scope =>
             {
-
                 value = ops.convert_to_tensor(value, name: "value");
                 Tensor size = array_ops.size(value, out_type: dtypes.int64);
-                Func<ITensorOrOperation> fu_true = () => math_ops.cast(_count_nonzero(value, dtype: dtypes.int32));
                 Tensor zero_fraction_float32 = null;
 
                 size = gen_math_ops.less_equal(size, dtypes.int32.max());