From 36ef4264270e3815ea4b485a52e35e26e43e2112 Mon Sep 17 00:00:00 2001
From: Yasuhiro Matsumoto <mattn.jp@gmail.com>
Date: Thu, 24 Oct 2019 13:06:50 +0900
Subject: [PATCH] Check nil returned

---
 tensorflow/go/graph.go | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/tensorflow/go/graph.go b/tensorflow/go/graph.go
index de355977b53..b3b2c9cc20a 100644
--- a/tensorflow/go/graph.go
+++ b/tensorflow/go/graph.go
@@ -124,11 +124,12 @@ func (g *Graph) ImportWithOptions(def []byte, options GraphImportOptions) error
 
 	buf := C.TF_NewBuffer()
 	defer C.TF_DeleteBuffer(buf)
-	// Would have preferred to use C.CBytes, but that does not play well
-	// with "go vet" till https://github.com/golang/go/issues/17201 is
-	// resolved.
 	buf.length = C.size_t(len(def))
 	buf.data = C.CBytes(def)
+	if buf.data == nil {
+		return fmt.Errorf("unable to allocate memory")
+	}
+	defer C.free(buf.data)
 
 	status := newStatus()