Searched refs:_BroadcastMul (Results 1 – 2 of 2) sorted by relevance
503 def _BroadcastMul(vec, mat): function527 grad = _BroadcastMul(grad_loss, softmax_grad)540 return grad, _BroadcastMul(grad_loss, -nn_ops.log_softmax(logits))553 grad = _BroadcastMul(grad_loss, softmax_grad)
44 from tensorflow.python.ops.nn_grad import _BroadcastMul255 return [_BroadcastMul(grad_loss, grad_without_gradient), None, None, None]