diff options
author | Sherry Moore <sherrym@google.com> | 2016-03-03 13:51:32 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-03-03 15:49:53 -0800 |
commit | fa07bfda6896748b747334e750763f6feea91129 (patch) | |
tree | ea67547b78445227172fb237d428de374469325e /tensorflow/core/framework/config.proto | |
parent | 778399465a28565360a496c750bab570216b5e87 (diff) |
Added timeout support for Session APIs.
To use in Python Session API:
sess = tf.Session(
config=tf.ConfigProto(session_timeout_in_ms=your_timeout_value))
Change: 116288637
Diffstat (limited to 'tensorflow/core/framework/config.proto')
-rw-r--r-- | tensorflow/core/framework/config.proto | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/core/framework/config.proto b/tensorflow/core/framework/config.proto index 7482d14d75..c6da869cce 100644 --- a/tensorflow/core/framework/config.proto +++ b/tensorflow/core/framework/config.proto @@ -131,6 +131,11 @@ message ConfigProto { // Options that apply to all graphs. GraphOptions graph_options = 10; + + // Global timeout for all blocking operations in this session. If non-zero, + // and not overridden on a per-operation basis, this value will be used as the + // deadline for all blocking operations. + int64 operation_timeout_in_ms = 11; }; // EXPERIMENTAL. Options for a single Run() call. @@ -140,6 +145,9 @@ message RunOptions { FULL_TRACE = 1; } TraceLevel trace_level = 1; + + // Time to wait for operation to complete in milliseconds. + int64 timeout_in_ms = 2; } // EXPERIMENTAL. Metadata output (i.e., non-Tensor) for a single Run() call. |