Searched refs:trainCfg (Results 1 - 3 of 3) sorted by relevance
| /third_party/mindspore/test/fuzztest/mindspore_fuzzer/ |
| H A D | mindspore_fuzzer.cpp | 267 auto trainCfg = OH_AI_TrainCfgCreate(); in MSModelFuzzTest_Null() local
268 if (trainCfg == nullptr) { in MSModelFuzzTest_Null()
272 OH_AI_TrainCfgDestroy(&trainCfg); in MSModelFuzzTest_Null()
273 auto retLossName = OH_AI_TrainCfgGetLossName(trainCfg, nullptr); in MSModelFuzzTest_Null()
278 OH_AI_TrainCfgSetLossName(trainCfg, nullptr, 0); in MSModelFuzzTest_Null()
279 auto retOptLevel = OH_AI_TrainCfgGetOptimizationLevel(trainCfg); in MSModelFuzzTest_Null()
284 OH_AI_TrainCfgSetOptimizationLevel(trainCfg, OH_AI_KAUTO); in MSModelFuzzTest_Null()
285 ret = OH_AI_TrainModelBuild(model, nullptr, 0, OH_AI_MODELTYPE_MINDIR, nullptr, trainCfg); in MSModelFuzzTest_Null()
290 ret = OH_AI_TrainModelBuildFromFile(model, nullptr, OH_AI_MODELTYPE_MINDIR, nullptr, trainCfg); in MSModelFuzzTest_Null()
|
| /third_party/mindspore/test/unittest/ |
| H A D | mindspore_unit_test.cpp | 451 OH_AI_TrainCfgHandle trainCfg = OH_AI_TrainCfgCreate(); in HWTEST() local
452 ASSERT_NE(trainCfg, nullptr); in HWTEST()
454 OH_AI_TrainCfgDestroy(&trainCfg); in HWTEST()
455 ASSERT_EQ(trainCfg, nullptr); in HWTEST()
464 OH_AI_TrainCfgHandle trainCfg = OH_AI_TrainCfgCreate(); in HWTEST() local
465 ASSERT_NE(trainCfg, nullptr); in HWTEST()
468 OH_AI_TrainCfgSetLossName(trainCfg, const_cast<const char **>(setLossName), set_train_cfg_loss_name.size()); in HWTEST()
471 char **getLossName = OH_AI_TrainCfgGetLossName(trainCfg, &getNum); in HWTEST()
472 printf("trainCfg loss name: "); in HWTEST()
485 OH_AI_TrainCfgDestroy(&trainCfg); in HWTEST()
494 OH_AI_TrainCfgHandle trainCfg = OH_AI_TrainCfgCreate(); HWTEST() local
[all...] |
| /third_party/mindspore/test/utils/ |
| H A D | model_utils.cpp | 243 OH_AI_TrainCfgHandle trainCfg = OH_AI_TrainCfgCreate(); in ModelTrain() local
251 ret = OH_AI_TrainModelBuild(model, graphBuf, size, OH_AI_MODELTYPE_MINDIR, context, trainCfg); in ModelTrain()
255 ret = OH_AI_TrainModelBuildFromFile(model, graphPath, OH_AI_MODELTYPE_MINDIR, context, trainCfg); in ModelTrain()
|
Completed in 4 milliseconds