CANN/ops-math PadV3Grad算子
PadV3Grad【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√Atlas 200I/500 A2 推理产品×Atlas 推理系列产品√Atlas 训练系列产品√功能说明算子功能根据不同条件调用不同的底层padgrad算子pad_v3_grad_replicate/pad_v3_grad_replication/pad_v4_grad/reflection_pad3d_grad。调用说明算子名称对应README调用条件circular_pad_gradcircular_pad_grad当输入模式mode“circular”时调用算子circular_pad_grad。pad_v4_gradpad_v4_grad当输入张量维度小于等于4输入模式mode为“reflect”时调用算子pad_v4_grad。pad_v3_grad_replicatepad_v3_grad_replicate当输入张量维度小于等于4输入模式mode为“edge”时调用算子pad_v3_grad_replicate。pad_v3_grad_replicationpad_v3_grad_replication当输入张量维度等于5输入模式mode为“edge”时调用算子pad_v3_grad_replication。reflection_pad3d_gradreflection_pad3d_grad当输入张量维度等于5输入模式mode为“reflect”时调用算子reflection_pad3d_grad。【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考