Home
last modified time | relevance | path

Searched defs:log_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.c26 LogSoftmaxStruct *log_softmax = (LogSoftmaxStruct *)cdata; in LogSoftmaxLastAxisRun() local
57 LogSoftmaxStruct *log_softmax = (LogSoftmaxStruct *)self; in LogSoftmaxResize() local
78 LogSoftmaxStruct *log_softmax = (LogSoftmaxStruct *)self; in LogSoftmaxCompute() local
106 LogSoftmaxStruct *log_softmax = (LogSoftmaxStruct *)malloc(sizeof(LogSoftmaxStruct)); in CreateLogSoftmax() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/ops_info/
Dlog_softmax_info_test.cc31 LogSoftmaxInfoPtr log_softmax; variable