Searched refs:sum_grad (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linalg_grad_test.py | 50 sum_grad = gradients_impl.gradients(reduced, batch_identity)[0]
51 self.assertAllClose(batch_identity, self.evaluate(sum_grad))
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | math_grad.cc | 686 auto sum_grad = SumGradHelper(scope, op, grad_inputs); in MeanGrad() local
700 Div(scope, sum_grad, Cast(scope, group_size, sum_grad.type()))); in MeanGrad()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_grad.py | 248 sum_grad = _SumGrad(op, grad)[0]
256 factor = constant_op.constant(factor, dtype=sum_grad.dtype)
262 return math_ops.truediv(sum_grad, math_ops.cast(factor, sum_grad.dtype)), None
|