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

Merge pull request #28768 from bioothod:master

Browse Source 
PiperOrigin-RevId: 248754577
This commit is contained in:
TensorFlower Gardener 2019-05-17 11:24:07 -07:00
commit 1d0dea1376
1 changed files with 3 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tensorflow/go/graph.go
View File

@ -112,17 +112,9 @@ func (g *Graph) ImportWithOptions(def []byte, options GraphImportOptions) error
C.TF_ImportGraphDefOptionsSetPrefix(opts, cprefix)
if len(options.Device) != 0 {
// TODO(ashankar): Remove this error and uncomment below
// when a release of the C library which includes
// https://github.com/tensorflow/tensorflow/commit/e0af5ac53e5a8ad9b07cdd5738c0a8e12f938c4e
// has been made.
// See https://github.com/tensorflow/tensorflow/issues/23257
return fmt.Errorf("GraphImportOptions.Device is only supported with the TensorFlow C library versions after 1.12 (or built from master). See https://github.com/tensorflow/tensorflow/issues/23257")
/*
cdev := C.CString(options.Device)
defer C.free(unsafe.Pointer(cdev))
C.TF_ImportGraphDefOptionsSetDefaultDevice(opts, cdev)
*/
cdev := C.CString(options.Device)
defer C.free(unsafe.Pointer(cdev))
C.TF_ImportGraphDefOptionsSetDefaultDevice(opts, cdev)
}
buf := C.TF_NewBuffer()