Lines Matching refs:model
160 // model build and predict
161 void ModelPredict(OH_AI_ModelHandle model, OH_AI_ContextHandle context, std::string modelName,
167 printf("==========Build model by graphBuf==========\n");
172 ret = OH_AI_ModelBuild(model, graphBuf, size, OH_AI_MODELTYPE_MINDIR, context);
175 printf("==========Build model==========\n");
176 ret = OH_AI_ModelBuildFromFile(model, graphPath, OH_AI_MODELTYPE_MINDIR, context);
178 printf("==========build model return code:%d\n", ret);
181 OH_AI_TensorHandleArray inputs = OH_AI_ModelGetInputs(model);
185 OH_AI_Status resize_ret = OH_AI_ModelResize(model, inputs, &shapeInfos, inputs.handle_num);
197 predictRet = OH_AI_ModelPredict(model, inputs, &outputs, beforeCallBack, afterCallBack);
200 predictRet = OH_AI_ModelPredict(model, inputs, &outputs, nullptr, nullptr);
207 OH_AI_ModelDestroy(&model);
212 void ModelPredict_ModelBuild(OH_AI_ModelHandle model, OH_AI_ContextHandle context, std::string modelName,
218 printf("==========Build model by graphBuf==========\n");
223 ret = OH_AI_ModelBuild(model, graphBuf, size, OH_AI_MODELTYPE_INVALID, context);
229 printf("==========Build model==========\n");
230 ret = OH_AI_ModelBuildFromFile(model, graphPath, OH_AI_MODELTYPE_INVALID, context);
235 printf("==========build model return code:%d\n", ret);
238 // model train build and predict
239 void ModelTrain(OH_AI_ModelHandle model, OH_AI_ContextHandle context, std::string modelName,
246 printf("==========Build model by graphBuf==========\n");
251 ret = OH_AI_TrainModelBuild(model, graphBuf, size, OH_AI_MODELTYPE_MINDIR, context, trainCfg);
254 printf("==========Build model==========\n");
255 ret = OH_AI_TrainModelBuildFromFile(model, graphPath, OH_AI_MODELTYPE_MINDIR, context, trainCfg);
257 printf("==========build model return code:%d\n", ret);
260 OH_AI_TensorHandleArray inputs = OH_AI_ModelGetInputs(model);
264 OH_AI_Status resize_ret = OH_AI_ModelResize(model, inputs, &shapeInfos, inputs.handle_num);
269 ret = OH_AI_ModelSetTrainMode(model, true);
275 ret = OH_AI_RunStep(model, beforeCallBack, afterCallBack);
278 ret = OH_AI_RunStep(model, nullptr, nullptr);