Home
last modified time | relevance | path

Searched defs:LogSoftmax (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dlog_softmax.h35 LogSoftmax() : PrimitiveC(kNameLogSoftmax) { InitIOName({"x"}, {"output"}); } in LogSoftmax() function
/third_party/mindspore/mindspore/_extends/graph_kernel/expanders/
Dlogsoftmax.py22 class LogSoftmax(Expander): class
/third_party/mindspore/tests/st/ops/gpu/
Dtest_logsoftmax_op.py43 class LogSoftmax(nn.Cell): class
/third_party/mindspore/tests/st/ops/cpu/
Dtest_logsoftmax_op.py45 class LogSoftmax(nn.Cell): class
/third_party/mindspore/tests/st/ops/graph_kernel/
Dtest_logsoftmax.py26 class LogSoftmax(nn.Cell): class
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dlog_softmax_fp32.c46 void LogSoftmax(const float *input_ptr, float *output_ptr, float *sum_data, const SoftmaxParameter … in LogSoftmax() function
/third_party/mindspore/mindspore/nn/layer/
Dactivation.py98 class LogSoftmax(Cell): class
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dtrt_op_converter.cc826 MS_TRT_CONVERTER_FUNC_REG(LogSoftmax) { in MS_TRT_CONVERTER_FUNC_REG() argument
/third_party/mindspore/tests/ut/cpp/transform/
Dconvert_test.cc441 TEST_F(TestConvert, LogSoftmax) { in TEST_F() argument
/third_party/mindspore/mindspore/ops/operations/
Dnn_ops.py290 class LogSoftmax(Primitive): class