Function: set_dynamic_hw_size
|
C Prototype |
aclError aclmdlSetDynamicHWSize(uint32_t modelId, aclmdlDataset *dataset, size_t index, uint64_t height, uint64_t width) |
|---|---|
|
Python Function |
ret = acl.mdl.set_dynamic_hw_size(model_id, dataset, index, height, width) |
|
Function Usage |
Sets the runtime image size for dynamic-shape model inference. |
|
Input Description |
model_id: int, model ID. You can obtain the model ID after the model is successfully loaded by calling the following APIs:
dataset: input data of a model. Data of type aclmdlDataset describes the input data for model inference, while data of type aclDataBuffer describes the input buffer size and address. For details, see aclmdlDataset. index: int, index of the input dynamic batch, obtained by calling acl.mdl.get_input_index_by_name. For dynamic batch and image size, the input name is fixed to ascend_mbatch_shape_data. For dynamic AIPP, the input name is fixed to ascend_dynamic_aipp_data. height: int, height to be set. width: int, width to be set. The configured image size (height and width) must be among the image size profiles set during model building. You can also call acl.mdl.get_dynamic_hw to obtain the number of image size profiles supported by a specified model and the width and height of each profile. |
|
Return Value |
ret: int, error code.
|
|
Restrictions |
None |
|
Reference |
For details about the API call sequence and example, see Dynamic Batch/Dynamic Image Size/Dynamic Dimension (Setting Multi-Dimension Profiles). |