diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-30 11:57:45 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-30 12:02:09 -0700 |
| commit | 5fa4e1ac928b0512b28e955c588c5a7eab2ea046 (patch) | |
| tree | 86b89e41ce56fc375ccafd482af83bb4021ad5ad /tensorflow/python/ops | |
| parent | 4ecce5aa64587afe1cd07ee4c92bbb5ce2cf85df (diff) | |
Parallel_for: fix converters for some ops that don't support broadcasting.
PiperOrigin-RevId: 215133508
Diffstat (limited to 'tensorflow/python/ops')
| -rw-r--r-- | tensorflow/python/ops/parallel_for/pfor.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/python/ops/parallel_for/pfor.py b/tensorflow/python/ops/parallel_for/pfor.py index e0f6d51881..83cbe64ff2 100644 --- a/tensorflow/python/ops/parallel_for/pfor.py +++ b/tensorflow/python/ops/parallel_for/pfor.py @@ -1987,14 +1987,12 @@ def _convert_cast(pfor_input): @RegisterPForWithArgs("Pow", math_ops.pow) @RegisterPForWithArgs("RealDiv", math_ops.divide) @RegisterPForWithArgs("Real", math_ops.real) -@RegisterPForWithArgs("ReciprocalGrad", math_ops.reciprocal_grad) @RegisterPForWithArgs("Reciprocal", math_ops.reciprocal) @RegisterPForWithArgs("Relu6", nn_ops.relu6) @RegisterPForWithArgs("Relu", nn_ops.relu) @RegisterPForWithArgs("RightShift", bitwise_ops.right_shift) @RegisterPForWithArgs("Rint", math_ops.rint) @RegisterPForWithArgs("Round", math_ops.round) -@RegisterPForWithArgs("RsqrtGrad", math_ops.rsqrt_grad) @RegisterPForWithArgs("Rsqrt", math_ops.rsqrt) @RegisterPForWithArgs("Selu", nn_ops.selu) @RegisterPForWithArgs("Sigmoid", math_ops.sigmoid) @@ -2003,7 +2001,6 @@ def _convert_cast(pfor_input): @RegisterPForWithArgs("Sin", math_ops.sin) @RegisterPForWithArgs("Softplus", nn_ops.softplus) @RegisterPForWithArgs("Softsign", nn_ops.softsign) -@RegisterPForWithArgs("SqrtGrad", math_ops.sqrt_grad) @RegisterPForWithArgs("Sqrt", math_ops.sqrt) @RegisterPForWithArgs("SquaredDifference", math_ops.squared_difference) @RegisterPForWithArgs("Square", math_ops.square) @@ -2095,6 +2092,9 @@ def _convert_biasaddgrad(pfor_input): @RegisterPForWithArgs("SoftplusGrad") @RegisterPForWithArgs("SoftsignGrad") @RegisterPForWithArgs("TanhGrad") +@RegisterPForWithArgs("SqrtGrad") +@RegisterPForWithArgs("RsqrtGrad") +@RegisterPForWithArgs("ReciprocalGrad") def _convert_grads(pfor_input, op_type, *args, **kw_args): del args del kw_args |