Skip to content

Commit 9296b3d

Browse files
committed
update median mapping
1 parent 58cd6aa commit 9296b3d

File tree

2 files changed

+1
-5
lines changed

2 files changed

+1
-5
lines changed

tester/base.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -482,8 +482,6 @@ def gen_paddle_input(self):
482482
self.paddle_args[3] = "gels"
483483
elif "driver" in self.paddle_kwargs:
484484
self.paddle_kwargs["driver"] = "gels"
485-
if self.api_config.api_name == "paddle.Tensor.median":
486-
self.paddle_kwargs["mode"] = "min"
487485

488486
if self.need_check_grad():
489487
if (self.api_config.api_name[-1] == "_" and self.api_config.api_name[-2:] != "__") or self.api_config.api_name == "paddle.Tensor.__setitem__":

tester/paddle_to_torch/mapping.json

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3100,9 +3100,7 @@
31003100
},
31013101
"paddle.Tensor.median": {
31023102
"torch_api": "torch.Tensor.median",
3103-
"paddle_torch_args_map": {
3104-
"keepdim": "keepdim"
3105-
}
3103+
"Rule": "MedianRule"
31063104
},
31073105
"paddle.Tensor.mm": {
31083106
"torch_api": "torch.Tensor.mm",

0 commit comments

Comments
 (0)