acldvppCHW2HWC
Applicability
Product |
Supported |
|---|---|
Atlas 350 Accelerator Card |
☓ |
√ |
|
√ |
|
☓ |
|
☓ |
|
☓ |
Function
Converts the shape of the input image from <C, H, W> to <H, W, C>.
Prototype
Each operator has two-phase API calls. First, acldvppCHW2HWCGetWorkspaceSize is called to calculate the required workspace size based on the computation process. Then, acldvppCHW2HWC is called to perform the computation. The two-phase APIs are as follows:
- First-phase API:
1acldvppStatus acldvppCHW2HWCGetWorkspaceSize(const aclTensor *self, aclTensor *out, uint64_t *workspaceSize, aclOpExecutor **executor)
- Second-phase API:
1acldvppStatus acldvppCHW2HWC(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)
acldvppCHW2HWCGetWorkspaceSize
- Parameters:
- self: input tensor of the operator. aclCreateTensor needs to be called to create data of the aclTensor type (stored on the device). The dataType of the input tensor can be UINT8 or FLOAT, and the format can be NCHW. Discontinuous tensors are not supported, in addition, N can be 1 or empty, and C can be 1 or 3 (1 indicates the input GRAY image, and 3 indicates the input RGB image).
- out: operator output tensor. aclCreateTensor needs to be called to create data of the aclTensor type (stored on the device). The data type of the output tensor can be UINT8 or FLOAT, and the format can be NHWC. Discontinuous tensors are not supported, in addition, N can be 1 or empty, and C can be 1 or 3 (1 indicates the input GRAY image, and 3 indicates the input RGB image). The data type of out must be the same as that of self. The C, H, and W values of out must be the same as those of self.
- workspaceSize: size of the workspace to be allocated on the device.
- executor: operator executor, containing the operator computation process.
- Returns:
The status code acldvppStatus is returned. For details, see acldvpp Return Codes.
acldvppCHW2HWC
- Parameters:
- workspace: The aclrtMalloc API needs to be called to allocate the device memory. The memory size is workspaceSize. The memory address output by the aclrtMalloc API is passed here.
- workspaceSize: The value must be the same as that obtained by calling acldvppCHW2HWCGetWorkspaceSize.
- executor: op executor, which contains the operator computation process and is the same as the executor of acldvppCHW2HWCGetWorkspaceSize.
- stream: stream for executing the task. You can reuse the created stream to save resources or call aclrtCreateStream to create a stream, and then pass the stream as an input parameter.
- Returns
The status code acldvppStatus is returned. For details, see acldvpp Return Codes.
Constraints
The supported image resolution ranges from 6 x 4 to 4096 x 8192.