Home
last modified time | relevance | path

Searched refs:softmax_kernel (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/
Dsoftmax_grad.cc72 auto softmax_kernel = reinterpret_cast<SoftmaxGradCPUKernel *>(cdata); in SoftmaxGradRun() local
73 auto error_code = softmax_kernel->Execute(task_id); in SoftmaxGradRun()
Dsoftmax_cross_entropy_with_logits.cc84 auto softmax_kernel = reinterpret_cast<SoftmaxCrossEntropyWithLogitsCPUKernel *>(cdata); in SoftmaxCrossEntropyWithLogitsRun() local
85 auto error_code = softmax_kernel->Execute(task_id); in SoftmaxCrossEntropyWithLogitsRun()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dsoftmax_int8.cc118 auto softmax_kernel = reinterpret_cast<SoftmaxInt8CPUKernel *>(cdata); in SoftmaxRun() local
119 auto error_code = softmax_kernel->DoSoftmax(task_id); in SoftmaxRun()