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

Enable lazy_remote_inputs_copy by default.

Browse Source 
PiperOrigin-RevId: 285468401
Change-Id: I49f7477dad816de3f4986c01c35e2121bfb00539
This commit is contained in:
Yujing Zhang 2019-12-13 14:09:14 -08:00 committed by TensorFlower Gardener
parent a3783f4716
commit 913e50445b
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/c/eager/c_api_internal.h
View File

@ -58,7 +58,7 @@ struct TFE_ContextOptions {
TFE_DEVICE_PLACEMENT_SILENT};
TFE_ContextMirroringPolicy mirroring_policy{TFE_MIRRORING_NONE};
// If true, lazily copy the remote inputs of a function to the target devices.
bool lazy_remote_inputs_copy = false;
bool lazy_remote_inputs_copy = true;
};
struct TFE_Context {

6
tensorflow/python/eager/context.py
View File

@ -405,7 +405,7 @@ class Context(object):
if execution_mode is None:
execution_mode = SYNC
self._default_is_async = execution_mode == ASYNC
self._lazy_remote_inputs_copy = False
self._lazy_remote_inputs_copy = None
self._server_def = server_def
self._collective_ops_server_def = None
self._collective_leader = None
@ -506,9 +506,9 @@ class Context(object):
opts, self._mirroring_policy)
if self._default_is_async == ASYNC:
pywrap_tensorflow.TFE_ContextOptionsSetAsync(opts, True)
if self._lazy_remote_inputs_copy:
if self._lazy_remote_inputs_copy is not None:
pywrap_tensorflow.TFE_ContextOptionsSetLazyRemoteInputsCopy(
opts, True)
opts, self._lazy_remote_inputs_copy)
context_handle = pywrap_tensorflow.TFE_NewContext(opts)
finally:
pywrap_tensorflow.TFE_DeleteContextOptions(opts)