昇腾社区首页
中文
注册

aclnnAdaptiveMaxPool2d

支持的产品型号

  • Atlas 训练系列产品。
  • Atlas A2训练系列产品/Atlas 800I A2推理产品。

接口原型

  • 每个算子分为undefined,必须先调用“aclnnAdaptiveMaxPool2dGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnAdaptiveMaxPool2d”接口执行计算。
  • aclnnStatus aclnnAdaptiveMaxPool2dGetWorkspaceSize(const aclTensor* self, const aclIntArray* outputSize, aclTensor* outputOut, aclTensor* indicesOut, uint64_t* workspaceSize, aclOpExecutor** executor)
  • aclnnStatus aclnnAdaptiveMaxPool2d(void* workspace, uint64_t workspaceSize, aclOpExecutor* executor, aclrtStream stream)

功能描述

  • 算子功能: 对输入self的feature_map进行最大值池化操作。与max_pool2d的区别在于,不需要feature_map的长、宽被kernel_size整除,只需指定outputSize大小,并按outputSize的大小来划分pooling区域。

  • Shape描述:

    • self: (N, C, Hin, Win) 或者 (C, Hin, Win) 或 (N, Hin, Win, C)
    • outputSize: (Hout, Wout)
    • outputOut: (N, C, Hout, Wout) 或者 (C, Hout, Wout) 或 (N, Hout, Wout,C)
    • indicesOut: (N, C, Hout, Wout) 或者 (C, Hout, Wout) 或 (N, Hout, Wout,C)

aclnnAdaptiveMaxPool2dGetWorkspaceSize

  • 参数说明:

    • self(const aclTensor*, 计算输入):: Device侧的aclTensor, 数据类型支持FLOAT16、FLOAT和DOUBLE,且数据类型与outputOut的数据类型一致。支持undefined, undefined支持NCHW、NCL、NHWC。NCL输入时按照(C, Hin, Win)处理。
    • outputSize(const aclIntArray *, 计算输入):host侧的aclIntArray, 数据类型为int数组, size大小为2。表示输出结果在H和W维度上的空间大小。
    • outputOut(aclTensor *, 计算输出):Device侧的aclTensor, 数据类型支持FLOAT16、FLOAT和DOUBLE, 且数据类型与self的数据类型一致,shape与indicesOut一致。支持undefined, undefined支持NCHW、NCL、NHWC。NCL输入时按照(C, Hout, Wout)处理。
    • indicesOut(aclTensor *,计算输出):Device侧的aclTensor, 表示outputOut元素在输入self中的索引位置。数据类型支持INT64, 且shape与outputOut一致。支持undefined, undefined支持NCHW、NCL、NHWC。NCL输入时按照(C, Hout, Wout)处理。
    • workspaceSize(uint64_t*, 出参): 返回需要在Device侧申请的workspace大小。
    • executor(aclOpExecutor**, 出参): 返回op执行器,包含了算子计算流程。
  • 返回值:

    aclnnStatus:返回状态码,具体参见undefined

    [object Object]

aclnnAdaptiveMaxPool2d

  • 参数说明:

    • workspace(void *, 入参): 在Device侧申请的workspace内存地址。
    • workspaceSize(uint64_t, 入参): 在Device侧申请的workspace大小,由第一段接口aclnnAdaptiveMaxPool2dGetWorkspaceSize获取。
    • executor(aclOpExecutor *, 入参): op执行器,包含了算子计算流程。
    • stream(aclrtStream, 入参): 指定执行任务的 AscendCL Stream流。
  • 返回值: aclnnStatus: 返回状态码,具体参见undefined

约束与限制

调用示例

示例代码如下,仅供参考,具体编译和执行过程请参考undefined

[object Object]