Home
last modified time | relevance | path

Searched defs:softmax_ (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dlog_softmax.h26 SoftmaxStruct softmax_; member
/third_party/mindspore/mindspore-src/source/tests/st/ops/
Dtest_ops_softmax.py24 def softmax_(x, axis): function