Searched refs:_softmax (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 2794 def _softmax(logits, compute_op, dim=-1, name=None): function
2900 return _softmax(logits, gen_nn_ops.softmax, axis, name)
2927 return _softmax(logits, gen_nn_ops.softmax, axis, name)
2957 return _softmax(logits, gen_nn_ops.log_softmax, axis, name)
2984 return _softmax(logits, gen_nn_ops.log_softmax, axis, name)
|
| D | nn_test.py | 98 def _softmax(self, x): member in SoftmaxTest
109 y_np = self._softmax(x_np)
|