Adding GetAttributeType method to query for Type attributes (migueldeicaza#353)

vladimiroster · migueldeicaza · commit 8acdb03efa57 · 2018-11-27T13:46:02.000-08:00
diff --git a/TensorFlowSharp/Tensorflow.cs b/TensorFlowSharp/Tensorflow.cs
@@ -2232,7 +2232,21 @@ public TFAttributeMetadata GetAttributeMetadata (string attrName, TFStatus statu
 		// extern void TF_OperationGetAttrType (TF_Operation *oper, const char *attr_name, TF_DataType *value, TF_Status *status);
 		[DllImport (NativeBinding.TensorFlowLibrary)]
 		static extern unsafe void TF_OperationGetAttrType (TF_Operation oper, string attr_name, TFDataType* value, TF_Status status);
-		// TODO:
+
+		/// <summary>
+		/// Query the operation for a type attribute
+		/// </summary>
+		/// <value>The value (type) of the attribute.</value>
+		public unsafe TFDataType GetAttributeType (string attrName, TFStatus status = null)
+		{
+			if (handle == IntPtr.Zero)
+				TFDisposable.ObjectDisposedException ();
+			var cstatus = TFStatus.Setup (status);
+			TFDataType type_value = new TFDataType ();
+			TF_OperationGetAttrType (handle, attrName, &type_value, cstatus.Handle);
+			cstatus.CheckMaybeRaise (status);
+			return type_value;
+		}
 
 		// extern void TF_OperationGetAttrTypeList (TF_Operation *oper, const char *attr_name, TF_DataType *values, int max_values, TF_Status *status);
 		[DllImport (NativeBinding.TensorFlowLibrary)]
