Searched refs:compute_op (Results 1 – 1 of 1) sorted by relevance
2794 def _softmax(logits, compute_op, dim=-1, name=None): argument2833 return compute_op(logits, name=name)2862 output = compute_op(logits)