Searched refs:weighted_grads (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/python/ops/ |
D | math_grad.py | 401 weighted_grads = math_ops.divide(grad, num_selected) 402 gathered_grads = array_ops.gather(weighted_grads, op.inputs[1]) 504 weighted_grads = math_ops.divide(grad, num_selected) 505 gathered_grads, _, _ = _GatherDropNegatives(weighted_grads, None,
|
/external/tensorflow/tensorflow/cc/gradients/ |
D | math_grad.cc | 1284 auto weighted_grads = Div(scope, grad, num_selected); in UnsortedSegmentMinOrMaxGrad() local 1285 auto gathered_grads = GatherDropNegatives(scope, weighted_grads, in UnsortedSegmentMinOrMaxGrad()
|