experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use C.CBytes() instead of C.malloc()

Browse Source
This commit is contained in:
Yasuhiro Matsumoto 2019-10-24 12:33:04 +09:00
parent 4a377f0d6b
commit 905d28af0e
No known key found for this signature in database
GPG Key ID: 622DE34DC490584B
1 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/go/graph.go
View File

@ -128,12 +128,7 @@ func (g *Graph) ImportWithOptions(def []byte, options GraphImportOptions) error
// with "go vet" till https://github.com/golang/go/issues/17201 is
// resolved.
buf.length = C.size_t(len(def))
buf.data = C.malloc(buf.length)
if buf.data == nil {
return fmt.Errorf("unable to allocate memory")
}
defer C.free(buf.data)
C.memcpy(buf.data, C.CBytes(def), buf.length)
buf.data = C.CBytes(def)
status := newStatus()