diff options
| author |  Anna R <annarev@google.com> | 2018-03-15 12:43:01 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-15 12:47:47 -0700 |
| commit | d696214ab7d1565850702959983fa189d0bfe50c (patch) | |
| tree | 39353e4613710493190730c8ceda927b093bbc80 /tensorflow/contrib/nn | |
| parent | 0f508d4de379e800ad7f990de08959bbd6fcabb5 (diff) | |
Remove underscore prefix from broadcast_gradient_args op.
PiperOrigin-RevId: 189229472
Diffstat (limited to 'tensorflow/contrib/nn')
| -rw-r--r-- | tensorflow/contrib/nn/python/ops/scaled_softplus.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tensorflow/contrib/nn/python/ops/scaled_softplus.py b/tensorflow/contrib/nn/python/ops/scaled_softplus.py index fcbfbc239c..7184ef2b66 100644 --- a/tensorflow/contrib/nn/python/ops/scaled_softplus.py +++ b/tensorflow/contrib/nn/python/ops/scaled_softplus.py @@ -30,9 +30,7 @@ def _reduce_and_reshape_grad(g, t): """Returns the gradient, sum-reduced and reshaped to `t`'s shape.""" shape = array_ops.shape(t) g_shape = array_ops.shape(g) - # pylint: disable=protected-access - bcast_dims, _ = gen_array_ops._broadcast_gradient_args(shape, g_shape) - # pylint: enable=protected-access + bcast_dims, _ = gen_array_ops.broadcast_gradient_args(shape, g_shape) return array_ops.reshape(math_ops.reduce_sum(g, bcast_dims), shape) |