diff --git a/python/torchair/_ge_concrete_graph/ge_converter/aten/clamp.py b/python/torchair/_ge_concrete_graph/ge_converter/aten/clamp.py index ffd2d0367eeeed82ce01cdc43d78bf7750a02f22..52884640f599b929f39ec94996faf65dcd00a98b 100644 --- a/python/torchair/_ge_concrete_graph/ge_converter/aten/clamp.py +++ b/python/torchair/_ge_concrete_graph/ge_converter/aten/clamp.py @@ -53,7 +53,7 @@ def clamp(self, max_value, min_value, meta_outputs): self = dtype_promote(self, target_dtype=meta_outputs.dtype) min_value = dtype_promote(min_value, target_dtype=meta_outputs.dtype) max_value = dtype_promote(max_value, target_dtype=meta_outputs.dtype) - return ge.ClipByValue(self, min_value, max_value) + return ge.ClipByValueV2(self, min_value, max_value) @register_fx_node_ge_converter(torch.ops.aten.clamp.Tensor) diff --git a/python/torchair/_ge_concrete_graph/ge_converter/aten/hardtanh.py b/python/torchair/_ge_concrete_graph/ge_converter/aten/hardtanh.py index c410dcb9c1052d0870804d3209d20ac67bd40b2c..687a733dd60992a7bb6f6ccc5bf27db5d8fc3acf 100644 --- a/python/torchair/_ge_concrete_graph/ge_converter/aten/hardtanh.py +++ b/python/torchair/_ge_concrete_graph/ge_converter/aten/hardtanh.py @@ -42,7 +42,7 @@ def conveter_aten_hardtanh_default( """NB: aten::hardtanh(Tensor self, Scalar min_val=-1, Scalar max_val=1) -> Tensor""" if min_val >= max_val: return ge.Fill(ge.Shape(self), ge.Cast(max_val, dst_type=self.dtype)) - return ge.ClipByValue(self, min_val, max_val) + return ge.ClipByValueV2(self, min_val, max_val) @register_fx_node_ge_converter(torch.ops.aten.hardtanh.out)