Tester: rename inputs to example_inputs (#3800)

oscarandersson8218 · facebook-github-bot · commit 8c70a50b1561 · 2024-06-05T08:03:22.000-07:00
Summary: Rename inputs to example_inputs to differentiate between what's used as test data and what's used as example/calibration data. Change-Id: Ibe6e65fd3f6ec14a8f27fe2270595a3835b1e038 Pull Request resolved: #3800 Reviewed By: kirklandsign Differential Revision: D58137948 Pulled By: digantdesai fbshipit-source-id: 5be9ce447966a495eee2bf019b5b44360c231ae6
diff --git a/backends/arm/test/misc/test_debug_feats.py b/backends/arm/test/misc/test_debug_feats.py
@@ -45,7 +45,7 @@ def _tosa_MI_pipeline(self, module: torch.nn.Module, dump_file=None):
         (
             ArmTester(
                 module,
-                inputs=module.get_inputs(),
+                example_inputs=module.get_inputs(),
                 compile_spec=common.get_tosa_compile_spec(),
             )
             .export()
@@ -59,7 +59,7 @@ def _tosa_BI_pipeline(self, module: torch.nn.Module, dump_file=None):
         (
             ArmTester(
                 module,
-                inputs=module.get_inputs(),
+                example_inputs=module.get_inputs(),
                 compile_spec=common.get_tosa_compile_spec(),
             )
             .quantize()
@@ -101,7 +101,7 @@ def test_numerical_diff_prints(self):
         tester = (
             ArmTester(
                 model,
-                inputs=model.get_inputs(),
+                example_inputs=model.get_inputs(),
                 compile_spec=common.get_tosa_compile_spec(),
             )
             .quantize()
diff --git a/backends/arm/test/models/test_mobilenet_v2_arm.py b/backends/arm/test/models/test_mobilenet_v2_arm.py
@@ -52,7 +52,7 @@ def test_mv2_tosa_MI(self):
         (
             ArmTester(
                 self.mv2,
-                inputs=self.model_inputs,
+                example_inputs=self.model_inputs,
                 compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
             )
             .export()
@@ -67,7 +67,7 @@ def test_mv2_tosa_BI(self):
         (
             ArmTester(
                 self.mv2,
-                inputs=self.model_inputs,
+                example_inputs=self.model_inputs,
                 compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
             )
             .quantize()
@@ -86,7 +86,7 @@ def test_mv2_u55_BI(self):
         (
             ArmTester(
                 self.mv2,
-                inputs=self.model_inputs,
+                example_inputs=self.model_inputs,
                 compile_spec=common.get_u55_compile_spec(permute_memory_to_nhwc=True),
             )
             .quantize()
diff --git a/backends/arm/test/ops/test_add.py b/backends/arm/test/ops/test_add.py
@@ -61,7 +61,7 @@ def _test_add_tosa_MI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(),
             )
             .export()
@@ -80,7 +80,7 @@ def _test_add_tosa_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(),
             )
             .quantize()
@@ -100,7 +100,7 @@ def _test_add_u55_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_u55_compile_spec(),
             )
             .quantize()
diff --git a/backends/arm/test/ops/test_avg_pool.py b/backends/arm/test/ops/test_avg_pool.py
@@ -49,7 +49,7 @@ def _test_avgpool2d_tosa_MI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
             )
             .export()
@@ -69,7 +69,7 @@ def _test_avgpool2d_tosa_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
             )
             .quantize()
@@ -90,7 +90,7 @@ def _test_avgpool2d_tosa_u55_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_u55_compile_spec(permute_memory_to_nhwc=True),
             )
             .quantize()
diff --git a/backends/arm/test/ops/test_batch_norm.py b/backends/arm/test/ops/test_batch_norm.py
@@ -530,7 +530,7 @@ def _test_batchnorm2d_tosa_MI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(),
             )
             .export()
@@ -561,7 +561,7 @@ def _test_batchnorm2d_no_stats_tosa_MI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(),
             )
             .export()
@@ -590,7 +590,7 @@ def _test_batchnorm2d_tosa_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(),
             )
             .quantize()
@@ -622,7 +622,7 @@ def _test_batchnorm2d_u55_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_u55_compile_spec(),
             )
             .quantize()
diff --git a/backends/arm/test/ops/test_clone.py b/backends/arm/test/ops/test_clone.py
@@ -37,7 +37,9 @@ def _test_clone_tosa_MI_pipeline(
     ):
         (
             ArmTester(
-                module, inputs=test_data, compile_spec=common.get_tosa_compile_spec()
+                module,
+                example_inputs=test_data,
+                compile_spec=common.get_tosa_compile_spec(),
             )
             .export()
             .check_count({"torch.ops.aten.clone.default": 1})
@@ -53,7 +55,9 @@ def _test_clone_tosa_BI_pipeline(
     ):
         (
             ArmTester(
-                module, inputs=test_data, compile_spec=common.get_tosa_compile_spec()
+                module,
+                example_inputs=test_data,
+                compile_spec=common.get_tosa_compile_spec(),
             )
             .quantize()
             .export()
@@ -70,7 +74,9 @@ def _test_clone_tosa_u55_pipeline(
     ):
         (
             ArmTester(
-                module, inputs=test_data, compile_spec=common.get_u55_compile_spec()
+                module,
+                example_inputs=test_data,
+                compile_spec=common.get_u55_compile_spec(),
             )
             .quantize()
             .export()
diff --git a/backends/arm/test/ops/test_conv.py b/backends/arm/test/ops/test_conv.py
@@ -247,7 +247,7 @@ def _test_conv2d_tosa_MI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
             )
             .export()
@@ -267,7 +267,7 @@ def _test_conv2d_tosa_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
             )
             .quantize()
@@ -286,7 +286,7 @@ def _test_conv2d_u55_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_u55_compile_spec(permute_memory_to_nhwc=True),
             )
             .quantize()
diff --git a/backends/arm/test/ops/test_conv_combos.py b/backends/arm/test/ops/test_conv_combos.py
@@ -160,7 +160,7 @@ def _test_conv_combo_tosa_MI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
             )
             .export()
@@ -182,7 +182,7 @@ def _test_conv_combo_tosa_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
             )
             .quantize()
@@ -201,7 +201,7 @@ def _test_conv_combo_u55_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_u55_compile_spec(permute_memory_to_nhwc=True),
             )
             .quantize()
diff --git a/backends/arm/test/ops/test_depthwise_conv.py b/backends/arm/test/ops/test_depthwise_conv.py
@@ -133,7 +133,7 @@ def _test_dw_conv2d_tosa_MI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
             )
             .export()
@@ -151,7 +151,7 @@ def _test_dw_conv2d_tosa_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
             )
             .quantize()
@@ -170,7 +170,7 @@ def _test_dw_conv2d_u55_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_u55_compile_spec(permute_memory_to_nhwc=True),
             )
             .quantize()
diff --git a/backends/arm/test/ops/test_div.py b/backends/arm/test/ops/test_div.py
@@ -107,7 +107,7 @@ def _test_div_tosa_MI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(),
             )
             .export()
@@ -126,7 +126,7 @@ def _test_div_tosa_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(),
             )
             .quantize()
@@ -146,7 +146,7 @@ def _test_div_u55_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_u55_compile_spec(),
             )
             .quantize()
diff --git a/backends/arm/test/ops/test_linear.py b/backends/arm/test/ops/test_linear.py
@@ -119,7 +119,7 @@ def _test_linear_tosa_MI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(),
             )
             .export()
@@ -138,7 +138,7 @@ def _test_linear_tosa_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(),
             )
             .quantize()
@@ -158,7 +158,7 @@ def _test_linear_tosa_u55_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_u55_compile_spec(),
             )
             .quantize()
diff --git a/backends/arm/test/ops/test_mean_dim.py b/backends/arm/test/ops/test_mean_dim.py
@@ -54,7 +54,7 @@ def _test_meandim_tosa_MI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
             )
             .export()
@@ -74,7 +74,7 @@ def _test_meandim_tosa_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
             )
             .quantize()
@@ -95,7 +95,7 @@ def _test_meandim_tosa_u55_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_u55_compile_spec(permute_memory_to_nhwc=True),
             )
             .quantize()
diff --git a/backends/arm/test/ops/test_softmax.py b/backends/arm/test/ops/test_softmax.py
@@ -42,7 +42,7 @@ def _test_softmax_tosa_MI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_tosa_compile_spec(),
             )
             .export()
@@ -61,7 +61,9 @@ def _test_softmax_tosa_BI_pipeline(
     ):
         (
             ArmTester(
-                module, inputs=test_data, compile_spec=common.get_tosa_compile_spec()
+                module,
+                example_inputs=test_data,
+                compile_spec=common.get_tosa_compile_spec(),
             )
             .quantize()
             .export()
@@ -81,7 +83,7 @@ def _test_softmax_tosa_u55_BI_pipeline(
         (
             ArmTester(
                 module,
-                inputs=test_data,
+                example_inputs=test_data,
                 compile_spec=common.get_u55_compile_spec(),
             )
             .quantize()
diff --git a/backends/arm/test/ops/test_view.py b/backends/arm/test/ops/test_view.py
diff --git a/backends/arm/test/tester/arm_tester.py b/backends/arm/test/tester/arm_tester.py
diff --git a/backends/xnnpack/test/tester/tester.py b/backends/xnnpack/test/tester/tester.py

Original file line number	Diff line number	Diff line change
`@@ -45,7 +45,7 @@ def _tosa_MI_pipeline(self, module: torch.nn.Module, dump_file=None):`
`45`	`45`	`(`
`46`	`46`	`ArmTester(`
`47`	`47`	`module,`
`48`		`- inputs=module.get_inputs(),`
	`48`	`+ example_inputs=module.get_inputs(),`
`49`	`49`	`compile_spec=common.get_tosa_compile_spec(),`
`50`	`50`	`)`
`51`	`51`	`.export()`
`@@ -59,7 +59,7 @@ def _tosa_BI_pipeline(self, module: torch.nn.Module, dump_file=None):`
`59`	`59`	`(`
`60`	`60`	`ArmTester(`
`61`	`61`	`module,`
`62`		`- inputs=module.get_inputs(),`
	`62`	`+ example_inputs=module.get_inputs(),`
`63`	`63`	`compile_spec=common.get_tosa_compile_spec(),`
`64`	`64`	`)`
`65`	`65`	`.quantize()`
`@@ -101,7 +101,7 @@ def test_numerical_diff_prints(self):`
`101`	`101`	`tester = (`
`102`	`102`	`ArmTester(`
`103`	`103`	`model,`
`104`		`- inputs=model.get_inputs(),`
	`104`	`+ example_inputs=model.get_inputs(),`
`105`	`105`	`compile_spec=common.get_tosa_compile_spec(),`
`106`	`106`	`)`
`107`	`107`	`.quantize()`
Original file line number	Diff line number	Diff line change
`@@ -52,7 +52,7 @@ def test_mv2_tosa_MI(self):`
`52`	`52`	`(`
`53`	`53`	`ArmTester(`
`54`	`54`	`self.mv2,`
`55`		`- inputs=self.model_inputs,`
	`55`	`+ example_inputs=self.model_inputs,`
`56`	`56`	`compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),`
`57`	`57`	`)`
`58`	`58`	`.export()`
`@@ -67,7 +67,7 @@ def test_mv2_tosa_BI(self):`
`67`	`67`	`(`
`68`	`68`	`ArmTester(`
`69`	`69`	`self.mv2,`
`70`		`- inputs=self.model_inputs,`
	`70`	`+ example_inputs=self.model_inputs,`
`71`	`71`	`compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),`
`72`	`72`	`)`
`73`	`73`	`.quantize()`
`@@ -86,7 +86,7 @@ def test_mv2_u55_BI(self):`
`86`	`86`	`(`
`87`	`87`	`ArmTester(`
`88`	`88`	`self.mv2,`
`89`		`- inputs=self.model_inputs,`
	`89`	`+ example_inputs=self.model_inputs,`
`90`	`90`	`compile_spec=common.get_u55_compile_spec(permute_memory_to_nhwc=True),`
`91`	`91`	`)`
`92`	`92`	`.quantize()`
Original file line number	Diff line number	Diff line change
`@@ -61,7 +61,7 @@ def _test_add_tosa_MI_pipeline(`
`61`	`61`	`(`
`62`	`62`	`ArmTester(`
`63`	`63`	`module,`
`64`		`- inputs=test_data,`
	`64`	`+ example_inputs=test_data,`
`65`	`65`	`compile_spec=common.get_tosa_compile_spec(),`
`66`	`66`	`)`
`67`	`67`	`.export()`
`@@ -80,7 +80,7 @@ def _test_add_tosa_BI_pipeline(`
`80`	`80`	`(`
`81`	`81`	`ArmTester(`
`82`	`82`	`module,`
`83`		`- inputs=test_data,`
	`83`	`+ example_inputs=test_data,`
`84`	`84`	`compile_spec=common.get_tosa_compile_spec(),`
`85`	`85`	`)`
`86`	`86`	`.quantize()`
`@@ -100,7 +100,7 @@ def _test_add_u55_BI_pipeline(`
`100`	`100`	`(`
`101`	`101`	`ArmTester(`
`102`	`102`	`module,`
`103`		`- inputs=test_data,`
	`103`	`+ example_inputs=test_data,`
`104`	`104`	`compile_spec=common.get_u55_compile_spec(),`
`105`	`105`	`)`
`106`	`106`	`.quantize()`
Original file line number	Diff line number	Diff line change
`@@ -49,7 +49,7 @@ def _test_avgpool2d_tosa_MI_pipeline(`
`49`	`49`	`(`
`50`	`50`	`ArmTester(`
`51`	`51`	`module,`
`52`		`- inputs=test_data,`
	`52`	`+ example_inputs=test_data,`
`53`	`53`	`compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),`
`54`	`54`	`)`
`55`	`55`	`.export()`
`@@ -69,7 +69,7 @@ def _test_avgpool2d_tosa_BI_pipeline(`
`69`	`69`	`(`
`70`	`70`	`ArmTester(`
`71`	`71`	`module,`
`72`		`- inputs=test_data,`
	`72`	`+ example_inputs=test_data,`
`73`	`73`	`compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),`
`74`	`74`	`)`
`75`	`75`	`.quantize()`
`@@ -90,7 +90,7 @@ def _test_avgpool2d_tosa_u55_BI_pipeline(`
`90`	`90`	`(`
`91`	`91`	`ArmTester(`
`92`	`92`	`module,`
`93`		`- inputs=test_data,`
	`93`	`+ example_inputs=test_data,`
`94`	`94`	`compile_spec=common.get_u55_compile_spec(permute_memory_to_nhwc=True),`
`95`	`95`	`)`
`96`	`96`	`.quantize()`
Original file line number	Diff line number	Diff line change
`@@ -530,7 +530,7 @@ def _test_batchnorm2d_tosa_MI_pipeline(`
`530`	`530`	`(`
`531`	`531`	`ArmTester(`
`532`	`532`	`module,`
`533`		`- inputs=test_data,`
	`533`	`+ example_inputs=test_data,`
`534`	`534`	`compile_spec=common.get_tosa_compile_spec(),`
`535`	`535`	`)`
`536`	`536`	`.export()`
`@@ -561,7 +561,7 @@ def _test_batchnorm2d_no_stats_tosa_MI_pipeline(`
`561`	`561`	`(`
`562`	`562`	`ArmTester(`
`563`	`563`	`module,`
`564`		`- inputs=test_data,`
	`564`	`+ example_example_inputs=test_data,`
`565`	`565`	`compile_spec=common.get_tosa_compile_spec(),`
`566`	`566`	`)`
`567`	`567`	`.export()`
`@@ -590,7 +590,7 @@ def _test_batchnorm2d_tosa_BI_pipeline(`
`590`	`590`	`(`
`591`	`591`	`ArmTester(`
`592`	`592`	`module,`
`593`		`- inputs=test_data,`
	`593`	`+ example_inputs=test_data,`
`594`	`594`	`compile_spec=common.get_tosa_compile_spec(),`
`595`	`595`	`)`
`596`	`596`	`.quantize()`
`@@ -622,7 +622,7 @@ def _test_batchnorm2d_u55_BI_pipeline(`
`622`	`622`	`(`
`623`	`623`	`ArmTester(`
`624`	`624`	`module,`
`625`		`- inputs=test_data,`
	`625`	`+ example_inputs=test_data,`
`626`	`626`	`compile_spec=common.get_u55_compile_spec(),`
`627`	`627`	`)`
`628`	`628`	`.quantize()`
Original file line number	Diff line number	Diff line change
`@@ -37,7 +37,9 @@ def _test_clone_tosa_MI_pipeline(`
`37`	`37`	`):`
`38`	`38`	`(`
`39`	`39`	`ArmTester(`
`40`		`- module, inputs=test_data, compile_spec=common.get_tosa_compile_spec()`
	`40`	`+ module,`
	`41`	`+ example_inputs=test_data,`
	`42`	`+ compile_spec=common.get_tosa_compile_spec(),`
`41`	`43`	`)`
`42`	`44`	`.export()`
`43`	`45`	`.check_count({"torch.ops.aten.clone.default": 1})`
`@@ -53,7 +55,9 @@ def _test_clone_tosa_BI_pipeline(`
`53`	`55`	`):`
`54`	`56`	`(`
`55`	`57`	`ArmTester(`
`56`		`- module, inputs=test_data, compile_spec=common.get_tosa_compile_spec()`
	`58`	`+ module,`
	`59`	`+ example_inputs=test_data,`
	`60`	`+ compile_spec=common.get_tosa_compile_spec(),`
`57`	`61`	`)`
`58`	`62`	`.quantize()`
`59`	`63`	`.export()`
`@@ -70,7 +74,9 @@ def _test_clone_tosa_u55_pipeline(`
`70`	`74`	`):`
`71`	`75`	`(`
`72`	`76`	`ArmTester(`
`73`		`- module, inputs=test_data, compile_spec=common.get_u55_compile_spec()`
	`77`	`+ module,`
	`78`	`+ example_inputs=test_data,`
	`79`	`+ compile_spec=common.get_u55_compile_spec(),`
`74`	`80`	`)`
`75`	`81`	`.quantize()`
`76`	`82`	`.export()`
Original file line number	Diff line number	Diff line change
`@@ -247,7 +247,7 @@ def _test_conv2d_tosa_MI_pipeline(`
`247`	`247`	`(`
`248`	`248`	`ArmTester(`
`249`	`249`	`module,`
`250`		`- inputs=test_data,`
	`250`	`+ example_inputs=test_data,`
`251`	`251`	`compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),`
`252`	`252`	`)`
`253`	`253`	`.export()`
`@@ -267,7 +267,7 @@ def _test_conv2d_tosa_BI_pipeline(`
`267`	`267`	`(`
`268`	`268`	`ArmTester(`
`269`	`269`	`module,`
`270`		`- inputs=test_data,`
	`270`	`+ example_inputs=test_data,`
`271`	`271`	`compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),`
`272`	`272`	`)`
`273`	`273`	`.quantize()`
`@@ -286,7 +286,7 @@ def _test_conv2d_u55_BI_pipeline(`
`286`	`286`	`(`
`287`	`287`	`ArmTester(`
`288`	`288`	`module,`
`289`		`- inputs=test_data,`
	`289`	`+ example_inputs=test_data,`
`290`	`290`	`compile_spec=common.get_u55_compile_spec(permute_memory_to_nhwc=True),`
`291`	`291`	`)`
`292`	`292`	`.quantize()`
Original file line number	Diff line number	Diff line change
`@@ -160,7 +160,7 @@ def _test_conv_combo_tosa_MI_pipeline(`
`160`	`160`	`(`
`161`	`161`	`ArmTester(`
`162`	`162`	`module,`
`163`		`- inputs=test_data,`
	`163`	`+ example_inputs=test_data,`
`164`	`164`	`compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),`
`165`	`165`	`)`
`166`	`166`	`.export()`
`@@ -182,7 +182,7 @@ def _test_conv_combo_tosa_BI_pipeline(`
`182`	`182`	`(`
`183`	`183`	`ArmTester(`
`184`	`184`	`module,`
`185`		`- inputs=test_data,`
	`185`	`+ example_inputs=test_data,`
`186`	`186`	`compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),`
`187`	`187`	`)`
`188`	`188`	`.quantize()`
`@@ -201,7 +201,7 @@ def _test_conv_combo_u55_BI_pipeline(`
`201`	`201`	`(`
`202`	`202`	`ArmTester(`
`203`	`203`	`module,`
`204`		`- inputs=test_data,`
	`204`	`+ example_inputs=test_data,`
`205`	`205`	`compile_spec=common.get_u55_compile_spec(permute_memory_to_nhwc=True),`
`206`	`206`	`)`
`207`	`207`	`.quantize()`
Original file line number	Diff line number	Diff line change
`@@ -133,7 +133,7 @@ def _test_dw_conv2d_tosa_MI_pipeline(`
`133`	`133`	`(`
`134`	`134`	`ArmTester(`
`135`	`135`	`module,`
`136`		`- inputs=test_data,`
	`136`	`+ example_inputs=test_data,`
`137`	`137`	`compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),`
`138`	`138`	`)`
`139`	`139`	`.export()`
`@@ -151,7 +151,7 @@ def _test_dw_conv2d_tosa_BI_pipeline(`
`151`	`151`	`(`
`152`	`152`	`ArmTester(`
`153`	`153`	`module,`
`154`		`- inputs=test_data,`
	`154`	`+ example_inputs=test_data,`
`155`	`155`	`compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),`
`156`	`156`	`)`
`157`	`157`	`.quantize()`
`@@ -170,7 +170,7 @@ def _test_dw_conv2d_u55_BI_pipeline(`
`170`	`170`	`(`
`171`	`171`	`ArmTester(`
`172`	`172`	`module,`
`173`		`- inputs=test_data,`
	`173`	`+ example_inputs=test_data,`
`174`	`174`	`compile_spec=common.get_u55_compile_spec(permute_memory_to_nhwc=True),`
`175`	`175`	`)`
`176`	`176`	`.quantize()`
Original file line number	Diff line number	Diff line change
`@@ -107,7 +107,7 @@ def _test_div_tosa_MI_pipeline(`
`107`	`107`	`(`
`108`	`108`	`ArmTester(`
`109`	`109`	`module,`
`110`		`- inputs=test_data,`
	`110`	`+ example_inputs=test_data,`
`111`	`111`	`compile_spec=common.get_tosa_compile_spec(),`
`112`	`112`	`)`
`113`	`113`	`.export()`
`@@ -126,7 +126,7 @@ def _test_div_tosa_BI_pipeline(`
`126`	`126`	`(`
`127`	`127`	`ArmTester(`
`128`	`128`	`module,`
`129`		`- inputs=test_data,`
	`129`	`+ example_inputs=test_data,`
`130`	`130`	`compile_spec=common.get_tosa_compile_spec(),`
`131`	`131`	`)`
`132`	`132`	`.quantize()`
`@@ -146,7 +146,7 @@ def _test_div_u55_BI_pipeline(`
`146`	`146`	`(`
`147`	`147`	`ArmTester(`
`148`	`148`	`module,`
`149`		`- inputs=test_data,`
	`149`	`+ example_inputs=test_data,`
`150`	`150`	`compile_spec=common.get_u55_compile_spec(),`
`151`	`151`	`)`
`152`	`152`	`.quantize()`