Home
last modified time | relevance | path

Searched refs:train_cfg (Results 1 - 2 of 2) sorted by relevance

/third_party/mindspore/interfaces/kits/c/
H A Dmodel.h176 * @brief Destroy the train_cfg object. Only valid for Lite Train.
177 * @param train_cfg TrainCfg object handle.
180 OH_AI_API void OH_AI_TrainCfgDestroy(OH_AI_TrainCfgHandle *train_cfg);
184 * @param train_cfg TrainCfg object handle.
189 OH_AI_API char **OH_AI_TrainCfgGetLossName(OH_AI_TrainCfgHandle train_cfg, size_t *num);
193 * @param train_cfg TrainCfg object handle.
198 OH_AI_API void OH_AI_TrainCfgSetLossName(OH_AI_TrainCfgHandle train_cfg, const char **loss_name, size_t num);
201 * @brief Obtains optimization level of the train_cfg. Only valid for Lite Train.
202 * @param train_cfg TrainCfg object handle.
206 OH_AI_API OH_AI_OptimizationLevel OH_AI_TrainCfgGetOptimizationLevel(OH_AI_TrainCfgHandle train_cfg);
[all...]
/third_party/mindspore/test/unittest/
H A Dmindspore_unit_test.cpp553 OH_AI_TrainCfgHandle train_cfg = OH_AI_TrainCfgCreate(); in HWTEST() local
554 ASSERT_NE(train_cfg, nullptr); in HWTEST()
557 context, train_cfg); in HWTEST()
596 OH_AI_TrainCfgHandle train_cfg = OH_AI_TrainCfgCreate(); in HWTEST() local
597 ASSERT_NE(train_cfg, nullptr); in HWTEST()
600 context, train_cfg); in HWTEST()
649 OH_AI_TrainCfgHandle train_cfg = OH_AI_TrainCfgCreate(); in HWTEST() local
650 ASSERT_NE(train_cfg, nullptr); in HWTEST()
653 context, train_cfg); in HWTEST()
720 OH_AI_TrainCfgHandle train_cfg in HWTEST() local
769 OH_AI_TrainCfgHandle train_cfg = OH_AI_TrainCfgCreate(); HWTEST() local
[all...]

Completed in 2 milliseconds