acldvppHWC2CHW

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 <H, W, C> to <C, H, W>.

Prototype

Each operator has two-phase API calls. First, acldvppHWC2CHWGetWorkspaceSize is called to calculate the required workspace size based on the computation process. Then, acldvppHWC2CHW is called to perform the computation. The two-phase APIs are as follows:

  • First-phase API:
    1
    acldvppStatus acldvppHWC2CHWGetWorkspaceSize(const aclTensor *self, aclTensor *out, uint64_t *workspaceSize, aclOpExecutor **executor)
    
  • Second-phase API:
    1
    acldvppStatus acldvppHWC2CHW(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)
    

acldvppHWC2CHWGetWorkspaceSize

  • 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 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).
    • out: output tensor of the operator. Call aclCreateTensor 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 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). The data type of of out must be the same as that of self. The values of C, H, and W 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.

acldvppHWC2CHW

  • 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 acldvppHWC2CHWGetWorkspaceSize.
    • executor: op executor, which contains the operator computation process and is the same as the executor of acldvppHWC2CHWGetWorkspaceSize.
    • 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.