HomeSort by relevance Sort by last modified time
    Searched refs:sum_grad (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/python/kernel_tests/
linalg_grad_test.py 51 sum_grad = gradients_impl.gradients(reduced, batch_identity)[0]
52 self.assertAllClose(batch_identity.eval(), sum_grad.eval())
  /external/tensorflow/tensorflow/cc/gradients/
math_grad.cc 715 auto sum_grad = SumGradHelper(scope, op, grad_inputs); local
727 // propagate sum_grad/group_size
729 Div(scope, sum_grad, Cast(scope, group_size, sum_grad.type())));
    [all...]
  /external/tensorflow/tensorflow/python/ops/
math_grad.py 99 sum_grad = _SumGrad(op, grad)[0]
107 factor = constant_op.constant(factor, dtype=sum_grad.dtype)
113 return math_ops.truediv(sum_grad, math_ops.cast(factor, sum_grad.dtype)), None
    [all...]

Completed in 233 milliseconds