CANN/Ascend C卷积3D切分接口

📅 2026/7/15 14:50:54
CANN/Ascend C卷积3D切分接口
GetTiling功能说明获取Tiling参数。函数原型int64_t GetTiling(optiling::TConv3DApiTiling tiling)int64_t GetTiling(AscendC::tiling::TConv3DApiTiling tiling)参数说明参数名输入/输出描述tiling输出Conv3D的Tiling结构体用于存储最终的Tiling结果。TConv3DApiTiling结构介绍请参考TConv3DApiTiling结构体。返回值说明如果返回值不为-1则代表Tiling计算成功用户可以使用该Tiling结构的值。如果返回值为-1则代表Tiling计算失败该Tiling结果无法使用。约束说明调用GetTiling接口前必须先调用SetOrgInputShape、SetOrgWeightShape、SetSingleWeightShape、SetSingleOutputShape。调用示例// 实例化Conv3d Api auto ascendcPlatform platform_ascendc::PlatformAscendC(context-GetPlatformInfo()); Conv3dTilingApi::Conv3dTiling conv3dApiTiling(ascendcPlatform ); conv3dApiTiling.SetOrgInputShape(orgCi, orgDi, orgHi, orgWi); conv3dApiTiling.SetOrgWeightShape(cout, kd, kh, kw); conv3dApiTiling.SetSingleWeightShape(singleCi, singleKd, singleKh, singleKw); conv3dApiTiling.SetSingleOutputShape(singleCo, singleDo, singleM); ... conv3dApiTiling.GetTiling(tilingData.conv3ApiTilingData);创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考