MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / _convert_softmax

Function _convert_softmax

tensorflow/python/ops/parallel_for/pfor.py:1725–1727  ·  view source on GitHub ↗
(pfor_input, op_type, op_func)

Source from the content-addressed store, hash-verified

1723@RegisterPForWithArgs("LogSoftmax", gen_nn_ops.log_softmax)
1724@RegisterPForWithArgs("Softmax", gen_nn_ops.softmax)
1725def _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

Callers

nothing calls this directly

Calls 2

stacked_inputMethod · 0.80
wrapFunction · 0.70

Tested by

no test coverage detected