acldvppCHW2HWC

Applicability

Product

Supported

Atlas 350 Accelerator Card

Atlas A3 training product/Atlas A3 inference product

Atlas A2 training product/Atlas A2 inference product

Atlas 200I/500 A2 inference product

Atlas inference product

Atlas training product

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:
    1
    acldvppStatus acldvppCHW2HWCGetWorkspaceSize(const aclTensor *self, aclTensor *out, uint64_t *workspaceSize, aclOpExecutor **executor)
    
  • Second-phase API:
    1
    acldvppStatus 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.