Searched refs:_wrap_2d_function (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 3577 def _wrap_2d_function(inputs, compute_op, dim=-1, name=None): function
3705 return _wrap_2d_function(logits, gen_nn_ops.softmax, axis, name)
3733 return _wrap_2d_function(logits, gen_nn_ops.softmax, axis, name)
3764 return _wrap_2d_function(logits, gen_nn_ops.log_softmax, axis, name)
3792 return _wrap_2d_function(logits, gen_nn_ops.log_softmax, axis, name)
5868 return _wrap_2d_function(inputs, compute_on_matrix, axis)
|