(pfor_input, op_type, op_func)
| 1723 | @RegisterPForWithArgs("LogSoftmax", gen_nn_ops.log_softmax) |
| 1724 | @RegisterPForWithArgs("Softmax", gen_nn_ops.softmax) |
| 1725 | def _convert_softmax(pfor_input, op_type, op_func): |
| 1726 | del op_type |
| 1727 | return wrap(op_func(pfor_input.stacked_input(0)), True) |
| 1728 | |
| 1729 | |
| 1730 | # array_ops |
nothing calls this directly
no test coverage detected