diff --git a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_l1_loss_test.yaml b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_l1_loss_test.yaml
index ba910423421..21b5709c828 100644
--- a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_l1_loss_test.yaml
+++ b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_l1_loss_test.yaml
@@ -5,7 +5,7 @@ test-list:
         start-shape: [1, 1, 32, 32]
         end-shape: [6, 12, 256, 256]
         interval: [1, 1, 32, 32]
-        num-dims: [2]
+        num-dims: [2,3]
         num-shapes: 2
         num-samples: 128
         args-sampling-strategy: "all"
diff --git a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_mse_loss_test.yaml b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_mse_loss_test.yaml
index e4bcba43a3c..06b0a0199c5 100644
--- a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_mse_loss_test.yaml
+++ b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_mse_loss_test.yaml
@@ -5,7 +5,7 @@ test-list:
         start-shape: [1, 1, 32, 32]
         end-shape: [6, 12, 256, 256]
         interval: [1, 1, 32, 32]
-        num-dims: [2]
+        num-dims: [2,3]
         num-shapes: 2
         num-samples: 128
         args-sampling-strategy: "all"
diff --git a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_l1_loss_test.yaml b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_l1_loss_test.yaml
index ba910423421..21b5709c828 100644
--- a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_l1_loss_test.yaml
+++ b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_l1_loss_test.yaml
@@ -5,7 +5,7 @@ test-list:
         start-shape: [1, 1, 32, 32]
         end-shape: [6, 12, 256, 256]
         interval: [1, 1, 32, 32]
-        num-dims: [2]
+        num-dims: [2,3]
         num-shapes: 2
         num-samples: 128
         args-sampling-strategy: "all"
diff --git a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_mse_loss_test.yaml b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_mse_loss_test.yaml
index e4bcba43a3c..06b0a0199c5 100644
--- a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_mse_loss_test.yaml
+++ b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_mse_loss_test.yaml
@@ -5,7 +5,7 @@ test-list:
         start-shape: [1, 1, 32, 32]
         end-shape: [6, 12, 256, 256]
         interval: [1, 1, 32, 32]
-        num-dims: [2]
+        num-dims: [2,3]
         num-shapes: 2
         num-samples: 128
         args-sampling-strategy: "all"