FilesLast commitLast update
2 months ago
1 month ago
4 months ago
24 days ago
24 days ago
24 days ago
29 days ago
1 month ago
README

PadV3Grad

产品支持情况

产品 是否支持
Ascend 950PR&950DT系列产品 √
Atlas A3系列产品 √
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_grad circular_pad_grad 当输入模式(mode)“circular”时调用算子circular_pad_grad。
pad_v4_grad pad_v4_grad 当输入张量维度小于等于4,输入模式(mode)为“reflect”时调用算子pad_v4_grad。
pad_v3_grad_replicate pad_v3_grad_replicate 当输入张量维度小于等于4,输入模式(mode)为“edge”时调用算子pad_v3_grad_replicate。
pad_v3_grad_replication pad_v3_grad_replication 当输入张量维度等于5,输入模式(mode)为“edge”时调用算子pad_v3_grad_replication。
reflection_pad3d_grad reflection_pad3d_grad 当输入张量维度等于5,输入模式(mode)为“reflect”时调用算子reflection_pad3d_grad。