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]