diff options
| author |  Andy Craze <accraze@gmail.com> | 2017-12-23 18:39:49 -0800 |
|---|---|---|
| committer | accraze <accraze@gmail.com> | 2018-05-05 08:58:26 -0700 |
| commit | 6c62fa93ad22594c65e1b3f7aecce79da660f5e9 (patch) | |
| tree | 75edb787ddfdb34700b9e8c8245b2c8e8d7640ea /tensorflow/contrib/framework | |
| parent | e49733b99ed9bedda46b32910cbd2183f12a4fe3 (diff) | |
update _DECORATED_OPS for each latest call
This fixes a race condition where `add_arg_scope` doesn't reliably extract the arg list.
fixes #11923
Diffstat (limited to 'tensorflow/contrib/framework')
| -rw-r--r-- | tensorflow/contrib/framework/python/ops/arg_scope.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/tensorflow/contrib/framework/python/ops/arg_scope.py b/tensorflow/contrib/framework/python/ops/arg_scope.py index 5b15033995..ebe5bade23 100644 --- a/tensorflow/contrib/framework/python/ops/arg_scope.py +++ b/tensorflow/contrib/framework/python/ops/arg_scope.py @@ -103,9 +103,8 @@ def _kwarg_names(func): def _add_op(op): - key = arg_scope_func_key(op) - if key not in _DECORATED_OPS: - _DECORATED_OPS[key] = _kwarg_names(op) + key_op = _key_op(op) + _DECORATED_OPS[key_op] = _kwarg_names(op) @tf_contextlib.contextmanager |