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

Address review comments and add some more API docs

Browse Source
This commit is contained in:
frreiss 2019-03-18 13:07:22 -07:00
parent 3b3397c609
commit aaec622b8a
1 changed files with 18 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
tensorflow/go/session.go
View File

@ -316,6 +316,11 @@ type SessionOptions struct {
// Config is a binary-serialized representation of the // Config is a binary-serialized representation of the
// tensorflow.ConfigProto protocol message // tensorflow.ConfigProto protocol message
// (https://www.tensorflow.org/code/tensorflow/core/protobuf/config.proto). // (https://www.tensorflow.org/code/tensorflow/core/protobuf/config.proto).
// You can populate this field in three ways. You can use the zero value
// of this field, which configures the session with a default set of
// options. Or you create a `Config` struct with your options and call that
// struct's `Bytes()` method. Or you can generate a byte string outside of
// Go and paste that string into your Go program as a string literal.
Config []byte Config []byte
} }
@ -355,15 +360,17 @@ func (o *SessionOptions) c() (ret *C.TF_SessionOptions, done func(), err error)
type JitLevel int type JitLevel int
const ( const (
// DEFAULT is the default setting for this version of TensorFlow, // JitDefault is the default setting for this version of TensorFlow
// Currently the default is OFF, but it will change to ON in a future // and corresponds to the "DEFAULT" level in the Python API.
// version. // Currently the default is `JitOff`, but it will change to `JitOn` in a
DEFAULT JitLevel = 0 // future version.
// OFF disables just-in-time compilation and will continue to do so JitDefault JitLevel = 0
// JitOff disables just-in-time compilation and will continue to do so
// even after JIT compilation is enabled by default. // even after JIT compilation is enabled by default.
OFF JitLevel = -1 JitOff JitLevel = -1
// ON is a synonym for the ON_1 optimization level. // JitOn enables just-in-time compilation. It is a synonym for the
ON JitLevel = 1 // "ON_1" optimization level in the Python API.
JitOn JitLevel = 1
) )
// Config represents session parameters as encoded in the tensorflow.ConfigProto // Config represents session parameters as encoded in the tensorflow.ConfigProto
@ -400,11 +407,11 @@ func (c *Config) Bytes() []byte {
// OptimizerOptions.global_jit_level in the Python API. // OptimizerOptions.global_jit_level in the Python API.
enableXLACompilationAsChar := C.uchar(0) enableXLACompilationAsChar := C.uchar(0)
switch c.GlobalJitLevel { switch c.GlobalJitLevel {
case DEFAULT: case JitDefault:
// TODO(frreiss): When the semantics of GlobalJitLevel.DEFAULT change to // TODO(frreiss): When the semantics of GlobalJitLevel.DEFAULT change to
// "on", uncomment the following line. // "on", uncomment the following line.
// enableXLACompilationAsChar = C.uchar(1) // enableXLACompilationAsChar = C.uchar(1)
case ON: case JitOn:
enableXLACompilationAsChar = C.uchar(1) enableXLACompilationAsChar = C.uchar(1)
} }
gpuMemoryAllowGrowthAsChar := C.uchar(0) gpuMemoryAllowGrowthAsChar := C.uchar(0)