as_base_type #136

Author: Oceania2018

src/TensorFlowNET.Core/Graphs/Graph.cs

@@ -3,7 +3,6 @@
 using System.Linq;
 using System.Runtime.InteropServices;
 using System.Text;
-using TF_DataType = Tensorflow.DataType;
 
 namespace Tensorflow
 {

src/TensorFlowNET.Core/Operations/OpDefLibrary.cs

@@ -72,12 +72,8 @@ public Operation _apply_op_helper(string op_type_name, string name = "", Diction
                         }
                         else
                         {
-                            var base_type = value.dtype;
-                            // base type
-                            if ((int)value.dtype > 100)
-                            {
-                                base_type = (TF_DataType)Enum.Parse(typeof(TF_DataType), ((int)value.dtype - 100).ToString());
-                            }
+                            var base_type = value.dtype.as_base_dtype();
+                            
                             input_types.Add(base_type);
                         }
                     }
@@ -151,7 +147,7 @@ public Operation _apply_op_helper(string op_type_name, string name = "", Diction
 
         public DataType _MakeType(TF_DataType v, AttrDef attr_def)
         {
-            return v.as_datatype_enum();
+            return v.as_base_dtype().as_datatype_enum();
         }
     }
 }

src/TensorFlowNET.Core/Tensors/dtypes.cs

@@ -24,6 +24,7 @@ public static Type as_numpy_datatype(this TF_DataType type)
                     throw new NotImplementedException("as_numpy_datatype failed");
             }
         }
+
         public static TF_DataType as_dtype(Type type)
         {
             TF_DataType dtype = TF_DataType.DtInvalid;
@@ -62,5 +63,12 @@ public static DataType as_datatype_enum(this TF_DataType type)
 
             return dtype;
         }
+
+        public static TF_DataType as_base_dtype(this TF_DataType type)
+        {
+            return (int)type > 100 ?
+                (TF_DataType)Enum.Parse(typeof(TF_DataType), ((int)type - 100).ToString()) :
+                type;
+        }
     }
 }

src/TensorFlowNET.Core/Variables/RefVariable.cs

@@ -16,7 +16,7 @@ public partial class RefVariable : VariableV1
 
         private Operation _initializer_op;
         public Operation initializer => _initializer_op;
-        public Operation op => _initializer_op;
+        public Operation op => _variable.op;
 
         public string name => _variable.name;
 
@@ -77,7 +77,7 @@ private void _init_from_args(object initial_value,
 
                     var shape = _initial_value.shape;
                     dtype = _initial_value.dtype;
-                    _variable = gen_state_ops.variable_v2(shape, dtype, name);
+                    _variable = gen_state_ops.variable_v2(shape, dtype.as_base_dtype(), name);
                 }
 
                 // Manually overrides the variable's shape with the initial value's.

test/TensorFlowNET.UnitTest/VariableTest.cs

@@ -29,7 +29,7 @@ public void ScalarVar()
         [TestMethod]
         public void Add()
         {
-            var x = tf.Variable(0, name: "x");
+            var x = tf.Variable(10, name: "x");
 
             var model = tf.global_variables_initializer();