Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/f16/
Darithmetic_compare_f16.c36 ArithmeticParameter *arithmetic_param = (ArithmeticParameter *)base->param_; in InitArithmeticCompareF16RunFunction() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/
Dtrain_populate_parameter.cc405 …ArithmeticParameter *arithmetic_param = reinterpret_cast<ArithmeticParameter *>(malloc(sizeof(Arit… in PopulateBiasGradParameter() local
473 …ArithmeticParameter *arithmetic_param = reinterpret_cast<ArithmeticParameter *>(malloc(sizeof(Arit… in PopulateArithmeticGradParameter() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/
Dopencl_fusion.cc330 auto *arithmetic_param = in TryMergeArithmeticAct() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/
Darithmetic_grad_fp32_tests.cc34 …ArithmeticParameter *arithmetic_param = static_cast<ArithmeticParameter *>(malloc(sizeof(Arithmeti… in PopulateArithmeticParameter() local