CANN/asc-devkit卷积Tiling接口

📅 2026/7/15 15:45:35
CANN/asc-devkit卷积Tiling接口
SetGradOutputType功能说明设置GradOutput的位置、数据格式、数据类型信息这些信息必须与Kernel侧的设置保持一致。函数原型void SetGradOutputType(ConvCommonApi::TPosition pos, ConvCommonApi::ConvFormat format, ConvCommonApi::ConvDtype dtype)参数说明表 1参数说明参数名输入/输出描述pos输入GradOutput在内存上的位置。当前仅支持TPosition::GM。format输入GradOutput的数据格式。当前仅支持ConvFormat::NDC1HWC0。dtype输入GradOutput的数据类型。当前仅支持ConvDtype::FLOAT16、ConvDtype::BF16。返回值说明无约束说明无调用示例optiling::Conv3DBackpropFilterTilingData tilingData; auto ascendcPlatform platform_ascendc::PlatformAscendCManager::GetInstance(); ConvBackpropApi::Conv3dBpFilterTiling conv3dBpDwTiling(*ascendcPlatform); conv3dBpDwTiling.SetGradOutputType(ConvCommonApi::TPosition::GM, ConvCommonApi::ConvFormat::NDC1HWC0, ConvCommonApi::ConvDtype::FLOAT16);创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考