Home
last modified time | relevance | path

Searched defs:sce_param (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/
Dsoftmax_crossentropy_fp32_tests.cc33 …auto sce_param = reinterpret_cast<SoftmaxCrossEntropyParameter *>(malloc(sizeof(SoftmaxCrossEntrop… in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32_grad/
Dsparse_softmax_cross_entropy_with_logits.cc83 auto sce_param = reinterpret_cast<SoftmaxCrossEntropyParameter *>(op_parameter_); in DoExecute() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/
Dtrain_populate_parameter.cc166 SoftmaxCrossEntropyParameter *sce_param = in PopulateSparseSoftmaxCrossEntropyWithLogitsParameter() local
182 SoftmaxCrossEntropyParameter *sce_param = in PopulateSoftmaxCrossEntropyParameter() local