model.h
Overview
Provides model-related APIs for model creation and inference.
Since:
9
Related Modules:
MindSpore
Summary
Structs
| Name | Description |
|---|
| OH_AI_TensorHandleArray | Defines the tensor array structure, which is used to store the tensor array pointer and tensor array length. |
| OH_AI_ShapeInfo | Defines dimension information. The maximum dimension is set by MS_MAX_SHAPE_NUM. |
| OH_AI_CallBackParam | Defines the operator information passed in a callback. |
Macros
| Name | Description |
|---|
| OH_AI_MAX_SHAPE_NUM 32 | Defines dimension information. The maximum dimension is set by MS_MAX_SHAPE_NUM. |
Types
Functions
| Name | Description |
|---|
| OH_AI_ModelCreate () | Creates a model object. |
| OH_AI_ModelDestroy (OH_AI_ModelHandle *model) | Destroys a model object. |
| OH_AI_ModelBuild (OH_AI_ModelHandle model, const void *model_data, size_t data_size, OH_AI_ModelType model_type, const OH_AI_ContextHandle model_context) | Loads and builds a MindSpore model from the memory buffer. |
| OH_AI_ModelBuildFromFile (OH_AI_ModelHandle model, const char *model_path, OH_AI_ModelType model_type, const OH_AI_ContextHandle model_context) | Loads and builds a MindSpore model from a model file. |
| OH_AI_ModelResize (OH_AI_ModelHandle model, const OH_AI_TensorHandleArray inputs, OH_AI_ShapeInfo *shape_infos, size_t shape_info_num) | Adjusts the input tensor shapes of a built model. |
| OH_AI_ModelPredict (OH_AI_ModelHandle model, const OH_AI_TensorHandleArray inputs, OH_AI_TensorHandleArray *outputs, const OH_AI_KernelCallBack before, const OH_AI_KernelCallBack after) | Performs model inference. |
| OH_AI_ModelGetInputs (const OH_AI_ModelHandle model) | Obtains the input tensor array structure of a model. |
| OH_AI_ModelGetOutputs (const OH_AI_ModelHandle model) | Obtains the output tensor array structure of a model. |
| OH_AI_ModelGetInputByTensorName (const OH_AI_ModelHandle model, const char *tensor_name) | Obtains the input tensor of a model by tensor name. |
| OH_AI_ModelGetOutputByTensorName (const OH_AI_ModelHandle model, const char *tensor_name) | Obtains the output tensor of a model by tensor name. |
你可能感兴趣的鸿蒙文章
harmony 鸿蒙Native APIs
harmony 鸿蒙AVCapability
harmony 鸿蒙AVDemuxer
harmony 鸿蒙AVMuxer
harmony 鸿蒙AVScreenCapture
harmony 鸿蒙AVSource
harmony 鸿蒙AudioDecoder
harmony 鸿蒙AudioEncoder
harmony 鸿蒙CodecBase
harmony 鸿蒙Core