Searched refs:shapeInfos (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/test/utils/ |
| D | model_utils.cpp | 162 … OH_AI_ShapeInfo shapeInfos, bool buildByGraph, bool isTranspose, bool isCallback) { in ModelPredict() argument
183 if (shapeInfos.shape_num != 0) { in ModelPredict()
185 OH_AI_Status resize_ret = OH_AI_ModelResize(model, inputs, &shapeInfos, inputs.handle_num); in ModelPredict()
213 OH_AI_ShapeInfo shapeInfos, bool buildByGraph, bool isTranspose, bool isCallback) { in ModelTrain() argument
235 if (shapeInfos.shape_num != 0) { in ModelTrain()
237 OH_AI_Status resize_ret = OH_AI_ModelResize(model, inputs, &shapeInfos, inputs.handle_num); in ModelTrain()
|
| D | model_utils.h | 44 OH_AI_ShapeInfo shapeInfos, bool buildByGraph, bool isTranspose, bool isCallback);
46 OH_AI_ShapeInfo shapeInfos, bool buildByGraph, bool isTranspose, bool isCallback);
|