@@ -1281,46 +1281,6 @@ def UnflattenIntNegativeOneSizeStaticModule_basic(module, tu: TestUtils):
12811281 module .forward (tu .rand (5 , 12 , 3 ))
12821282
12831283
1284- class UnflattenIntDynamicModule (torch .nn .Module ):
1285- def __init__ (self ):
1286- super ().__init__ ()
1287-
1288- @export
1289- @annotate_args (
1290- [
1291- None ,
1292- ([- 1 , 12 ], torch .float32 , True ),
1293- ]
1294- )
1295- def forward (self , inputs ):
1296- return torch .ops .aten .unflatten (inputs , 1 , [3 , 4 ])
1297-
1298-
1299- @register_test_case (module_factory = lambda : UnflattenIntDynamicModule ())
1300- def UnflattenIntDynamicModule_basic (module , tu : TestUtils ):
1301- module .forward (tu .rand (2 , 12 ))
1302-
1303-
1304- class UnflattenIntDynamicWithInferredSizeModule (torch .nn .Module ):
1305- def __init__ (self ):
1306- super ().__init__ ()
1307-
1308- @export
1309- @annotate_args (
1310- [
1311- None ,
1312- ([- 1 , 20 ], torch .float32 , True ),
1313- ]
1314- )
1315- def forward (self , inputs ):
1316- return torch .ops .aten .unflatten (inputs , 1 , [4 , - 1 ])
1317-
1318-
1319- @register_test_case (module_factory = lambda : UnflattenIntDynamicWithInferredSizeModule ())
1320- def UnflattenIntDynamicWithInferredSizeModule_basic (module , tu : TestUtils ):
1321- module .forward (tu .rand (3 , 20 ))
1322-
1323-
13241284# ==============================================================================
13251285
13261286
0 commit comments