@@ -2725,7 +2725,9 @@ def forward(self, input):
27252725 @tvm .script .ir_module
27262726 class expected1 :
27272727 @R .function
2728- def main (input_1 : R .Tensor ((1 , 3 , 10 , 10 ), dtype = "float32" )) -> R .Tuple (
2728+ def main (
2729+ input_1 : R .Tensor ((1 , 3 , 10 , 10 ), dtype = "float32" )
2730+ ) -> R .Tuple (
27292731 R .Tensor ((1 , 1 , 10 , 10 ), dtype = "float32" ),
27302732 R .Tensor ((1 , 1 , 10 , 10 ), dtype = "float32" ),
27312733 R .Tensor ((1 , 1 , 10 , 10 ), dtype = "float32" ),
@@ -2784,7 +2786,9 @@ def forward(self, data):
27842786 @tvm .script .ir_module
27852787 class expected1 :
27862788 @R .function
2787- def main (input_1 : R .Tensor ((3 , 3 , 10 , 10 ), dtype = "float32" )) -> R .Tuple (
2789+ def main (
2790+ input_1 : R .Tensor ((3 , 3 , 10 , 10 ), dtype = "float32" )
2791+ ) -> R .Tuple (
27882792 R .Tensor ((3 , 10 , 10 ), dtype = "float32" ),
27892793 R .Tensor ((3 , 10 , 10 ), dtype = "float32" ),
27902794 R .Tensor ((3 , 10 , 10 ), dtype = "float32" ),
@@ -2819,7 +2823,9 @@ def main(input_1: R.Tensor((3, 3, 10, 10), dtype="float32")) -> R.Tuple(
28192823 @tvm .script .ir_module
28202824 class expected2 :
28212825 @R .function
2822- def main (input_1 : R .Tensor ((3 , 3 , 10 , 10 ), dtype = "float32" )) -> R .Tuple (
2826+ def main (
2827+ input_1 : R .Tensor ((3 , 3 , 10 , 10 ), dtype = "float32" )
2828+ ) -> R .Tuple (
28232829 R .Tensor ((3 , 10 , 10 ), dtype = "float32" ),
28242830 R .Tensor ((3 , 10 , 10 ), dtype = "float32" ),
28252831 R .Tensor ((3 , 10 , 10 ), dtype = "float32" ),
@@ -2888,7 +2894,9 @@ def forward(self, input):
28882894 @tvm .script .ir_module
28892895 class Expected :
28902896 @R .function
2891- def main (input_1 : R .Tensor ((1 , 3 , 10 , 10 ), dtype = "float32" )) -> R .Tuple (
2897+ def main (
2898+ input_1 : R .Tensor ((1 , 3 , 10 , 10 ), dtype = "float32" )
2899+ ) -> R .Tuple (
28922900 R .Tensor ((1 , 1 , 10 , 10 ), dtype = "float32" ),
28932901 R .Tensor ((1 , 1 , 10 , 10 ), dtype = "float32" ),
28942902 R .Tensor ((1 , 1 , 10 , 10 ), dtype = "float32" ),
0 commit comments