三亩地 三亩地SAN MU DI · CODE DIARY
ARTICLE DETAIL

日记详情

真实记录编程学习的某一天,欢迎挑你感兴趣的翻一翻。

CANN Ascend C SetDilation函数

CANN Ascend C SetDilation函数

SetDilation

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言,原生支持C和C++标准规范,主要由类库和语言扩展层构成,提供多层级API,满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit

功能说明

设置Dilation信息,即卷积核Depth/Height/Width方向的扩张大小。

函数原型

void SetDilation(int64_t dilationD, int64_t dilationH, int64_t dilationW)

参数说明

表 1参数说明

参数名

输入/输出

描述

dilationD

输入

卷积核Weight的Depth方向扩张大小。

dilationH

输入

卷积核Weight的Height方向扩张大小。

dilationW

输入

卷积核Weight的Width方向扩张大小。

返回值说明

无

约束说明

无

调用示例

optiling::Conv3DBackpropFilterTilingData tilingData; auto ascendcPlatform = platform_ascendc::PlatformAscendCManager::GetInstance(); ConvBackpropApi::Conv3dBpFilterTiling conv3dBpDwTiling(*ascendcPlatform); conv3dBpDwTiling.SetDilation(dilationD, dilationH, dilationW);

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言,原生支持C和C++标准规范,主要由类库和语言扩展层构成,提供多层级API,满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

← 返回列表