diff --git a/mindspore/ops/operations/_quant_ops.py b/mindspore/ops/operations/_quant_ops.py index b960bb3605..f06746523c 100644 --- a/mindspore/ops/operations/_quant_ops.py +++ b/mindspore/ops/operations/_quant_ops.py @@ -44,7 +44,12 @@ __all__ = ["MinMaxUpdatePerLayer", "BatchNormFold2_D", "BatchNormFold2GradD", "BatchNormFold2GradReduce", - "IFMR" + "IFMR", + "ActsULQ", + "ActsULQInputGrad", + "ActULQClampMinGrad", + "ActULQClampMaxGrad", + "WtsARQ" ] @@ -1300,7 +1305,7 @@ class ActULQClampMinGrad(PrimitiveWithInfer): return tuple(output_shape) def infer_dtype(self, input_x, input_y, input_z): - return input_x + return mstype.float32 class ActULQClampMaxGrad(PrimitiveWithInfer): @@ -1337,7 +1342,7 @@ class ActULQClampMaxGrad(PrimitiveWithInfer): return tuple(output_shape) def infer_dtype(self, input_x, input_y, input_z): - return input_x + return mstype.float32 class WtsARQ(PrimitiveWithInfer):