From 6457a4e8e0105b2d77bb286a046223fa7dfbab80 Mon Sep 17 00:00:00 2001 From: Oceania2018 Date: Sat, 3 Aug 2019 01:01:33 -0500 Subject: [PATCH] exception when OpType handle is zero. --- src/TensorFlowNET.Core/Operations/Operation.cs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/TensorFlowNET.Core/Operations/Operation.cs b/src/TensorFlowNET.Core/Operations/Operation.cs index 8c4ce606..d7590b97 100644 --- a/src/TensorFlowNET.Core/Operations/Operation.cs +++ b/src/TensorFlowNET.Core/Operations/Operation.cs @@ -56,8 +56,8 @@ namespace Tensorflow public TF_DataType dtype => TF_DataType.DtInvalid; public string name => _handle == IntPtr.Zero ? null : c_api.StringPiece(c_api.TF_OperationName(_handle)); - public string OpType => c_api.StringPiece(c_api.TF_OperationOpType(_handle)); - public string Device => c_api.StringPiece(c_api.TF_OperationDevice(_handle)); + public string OpType => _handle == IntPtr.Zero ? null : c_api.StringPiece(c_api.TF_OperationOpType(_handle)); + public string Device => _handle == IntPtr.Zero ? null : c_api.StringPiece(c_api.TF_OperationDevice(_handle)); private NodeDef _node_def; public NodeDef node_def