Home
last modified time | relevance | path

Searched defs:ForwardPostExecute (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/
Dsoftmax_cross_entropy_with_logits.c20 void ForwardPostExecute(const float *labels, const float *logits, float *grads, float *output2, in ForwardPostExecute() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/mkldnn/
Dsoftmax_cross_entropy_with_logits_cpu_kernel.cc61 void SoftmaxCrossEntropyWithLogitsCpuKernelMod::ForwardPostExecute(const float *input0, const float… in ForwardPostExecute() function in mindspore::kernel::SoftmaxCrossEntropyWithLogitsCpuKernelMod
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32_grad/
Dsparse_softmax_cross_entropy_with_logits.cc33 int SparseSoftmaxCrossEntropyWithLogitsCPUKernel::ForwardPostExecute(const int *labels, const float… in ForwardPostExecute() function in mindspore::kernel::SparseSoftmaxCrossEntropyWithLogitsCPUKernel
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dsparse_softmax_cross_entropy_with_logits_cpu_kernel.cc85 void SparseSoftmaxCrossEntropyWithLogitsCpuKernelMod::ForwardPostExecute(const S *labels, const T *… in ForwardPostExecute() function in mindspore::kernel::SparseSoftmaxCrossEntropyWithLogitsCpuKernelMod