@@ -330,19 +330,26 @@ def tmp(t):
330
330
('kthvalue' , small_3d_unique , lambda t : [3 , - 1 ], 'neg_dim' ),
331
331
('lerp' , small_3d , lambda t : [small_3d (t ), 0.3 ],'' , types , False , "skipIfHalfTensor" ),
332
332
('max' , small_3d_unique , lambda t : [],'' , types , False , "skipIfHalfTensor" ),
333
- ('max' , small_3d_unique , lambda t : [1 ], 'dim' ),
334
- ('max' , small_3d_unique , lambda t : [- 1 ], 'neg_dim' ),
333
+ ('max' , small_3d_unique , lambda t : [1 ], 'dim' , types , False ,
334
+ "skipIfByteTensor;skipIfCharTensor;skipIfDoubleTensor;skipIfFloatTensor;skipIfHalfTensor;skipIfIntTensor;skipIfLongTensor;skipIfShortTensor" ),
335
+ ('max' , small_3d_unique , lambda t : [- 1 ], 'neg_dim' , types , False ,
336
+ "skipIfByteTensor;skipIfCharTensor;skipIfDoubleTensor;skipIfFloatTensor;skipIfHalfTensor;skipIfIntTensor;skipIfLongTensor;skipIfShortTensor" ),
335
337
('max' , medium_2d , lambda t : [medium_2d (t )], 'elementwise' ),
336
338
('min' , small_3d_unique , lambda t : [],'' , types , False , "skipIfHalfTensor" ),
337
- ('min' , small_3d_unique , lambda t : [1 ], 'dim' ),
338
- ('min' , small_3d_unique , lambda t : [- 1 ], 'neg_dim' ),
339
+ ('min' , small_3d_unique , lambda t : [1 ], 'dim' , types , False ,
340
+ "skipIfByteTensor;skipIfCharTensor;skipIfDoubleTensor;skipIfFloatTensor;skipIfHalfTensor;skipIfIntTensor;skipIfLongTensor;skipIfShortTensor" ),
341
+ ('min' , small_3d_unique , lambda t : [- 1 ], 'neg_dim' , types , False ,
342
+ "skipIfByteTensor;skipIfCharTensor;skipIfDoubleTensor;skipIfFloatTensor;skipIfHalfTensor;skipIfIntTensor;skipIfLongTensor;skipIfShortTensor" ),
339
343
('min' , medium_2d , lambda t : [medium_2d (t )], 'elementwise' ),
340
344
('mean' , small_3d , lambda t : [], '' , types , False , "skipIfHalfTensor" ),
341
345
('mean' , small_3d , lambda t : [- 1 ], 'neg_dim' , types , False , "skipIfDoubleTensor;skipIfFloatTensor;skipIfHalfTensor" ),
342
346
('mean' , small_3d , lambda t : [1 ], 'dim' , types , False , "skipIfDoubleTensor;skipIfFloatTensor;skipIfHalfTensor" ),
343
- ('mode' , small_3d , lambda t : [],),
344
- ('mode' , small_3d , lambda t : [1 ], 'dim' ),
345
- ('mode' , small_3d , lambda t : [- 1 ], 'neg_dim' ),
347
+ ('mode' , small_3d , lambda t : [],'' , types , False ,
348
+ "skipIfByteTensor;skipIfCharTensor;skipIfDoubleTensor;skipIfFloatTensor;skipIfHalfTensor;skipIfIntTensor;skipIfLongTensor;skipIfShortTensor" ),
349
+ ('mode' , small_3d , lambda t : [1 ], 'dim' , types , False ,
350
+ "skipIfByteTensor;skipIfCharTensor;skipIfDoubleTensor;skipIfFloatTensor;skipIfHalfTensor;skipIfIntTensor;skipIfLongTensor;skipIfShortTensor" ),
351
+ ('mode' , small_3d , lambda t : [- 1 ], 'neg_dim' , types , False ,
352
+ "skipIfByteTensor;skipIfCharTensor;skipIfDoubleTensor;skipIfFloatTensor;skipIfHalfTensor;skipIfIntTensor;skipIfLongTensor;skipIfShortTensor" ),
346
353
('mvlgamma' , lambda t : tensor_clamp (small_2d (t ), 0.1 , 10 ), lambda t : [1 ], '2d_p=1' , float_types_no_half , False , "skipIfDoubleTensor;skipIfFloatTensor" ),
347
354
('mvlgamma' , lambda t : tensor_clamp (small_2d (t ), 0.6 , 10 ), lambda t : [2 ], '2d_p=2' , float_types_no_half , False , "skipIfDoubleTensor;skipIfFloatTensor" ),
348
355
('remainder' , small_3d , lambda t : [3 ], 'value' , types , False , "skipIfHalfTensor" ),
0 commit comments