diff options
| author |  Yong Tang <yong.tang.github@outlook.com> | 2018-06-02 22:11:20 +0000 |
|---|---|---|
| committer | Yong Tang <yong.tang.github@outlook.com> | 2018-06-02 22:11:20 +0000 |
| commit | 13ceff2d4096554f195a3c865c1391500e172485 (patch) | |
| tree | 8d09d385f419e5e5e41d0a38c7f2b3d19a6ccc50 /tensorflow/contrib/constrained_optimization | |
| parent | 744cf3d3e06fb63ffa40086766137daedc01a5ba (diff) | |
Fix warning in constrained_optimization test
In constrained_optimization test, keep_dims was
used for reduce_sum. Since keep_dims has been deprecated
it generates unnecessary warning. This fix updates
keep_dims -> keepdims to disable the warning.
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
Diffstat (limited to 'tensorflow/contrib/constrained_optimization')
| -rw-r--r-- | tensorflow/contrib/constrained_optimization/python/swap_regret_optimizer.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/constrained_optimization/python/swap_regret_optimizer.py b/tensorflow/contrib/constrained_optimization/python/swap_regret_optimizer.py index 04014ab4ae..91b2486393 100644 --- a/tensorflow/contrib/constrained_optimization/python/swap_regret_optimizer.py +++ b/tensorflow/contrib/constrained_optimization/python/swap_regret_optimizer.py @@ -169,8 +169,8 @@ def _project_stochastic_matrix_wrt_euclidean_norm(matrix): del old_inactive # Needed by the condition, but not the body. iteration += 1 scale = (1.0 - standard_ops.reduce_sum( - matrix, axis=0, keep_dims=True)) / standard_ops.maximum( - 1.0, standard_ops.reduce_sum(inactive, axis=0, keep_dims=True)) + matrix, axis=0, keepdims=True)) / standard_ops.maximum( + 1.0, standard_ops.reduce_sum(inactive, axis=0, keepdims=True)) matrix += scale * inactive new_inactive = standard_ops.to_float(matrix > 0) matrix *= new_inactive |