|
|
|
@@ -15,7 +15,7 @@ |
|
|
|
"""HSwishGrad op""" |
|
|
|
from mindspore.ops.op_info_register import op_info_register, AkgRegOp, DataType |
|
|
|
|
|
|
|
hswishgrad_op_info = AkgRegOp("HSwishGrad") \ |
|
|
|
hswish_grad_op_info = AkgRegOp("HSwishGrad") \ |
|
|
|
.fusion_type("OPAQUE") \ |
|
|
|
.input(0, "y_grad") \ |
|
|
|
.input(1, "x") \ |
|
|
|
@@ -25,7 +25,7 @@ hswishgrad_op_info = AkgRegOp("HSwishGrad") \ |
|
|
|
.get_op_info() |
|
|
|
|
|
|
|
|
|
|
|
@op_info_register(hsigmoidgrad_op_info) |
|
|
|
@op_info_register(hswish_grad_op_info) |
|
|
|
def _hswish_grad_akg(): |
|
|
|
"""HSwishGrad AutoDiff register""" |
|
|
|
return |