IVGCVSW-7785 Extend support for 3D tensors BATCH_TO_SPACE and SPACE_TO_BATCH in CpuRef

* Both layers were assuming 4D tensors, now 3D is supported too.
* Remove some unnecessary includes
* Add Unit Tests

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I7bdd11e4936a27cd97ec65fd915e6ccaa1494cff

1 files changed, 0 insertions, 14 deletions

diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp
index cbc6723dbc..81e5c837a5 100644
--- a/src/backends/reference/RefLayerSupport.cpp
+++ b/src/backends/reference/RefLayerSupport.cpp
@@ -794,20 +794,6 @@ bool RefLayerSupport::IsBatchToSpaceNdSupported(const TensorInfo& input,
     supported &= CheckSupportRule(TypesAreEqual(input, output), reasonIfUnsupported,
                                   "Reference BatchToSpaceNd: input and output types mismatched.");
 
-    supported &= CheckSupportRule(TensorNumDimensionsAreCorrect(output, 4),
-                                  reasonIfUnsupported,
-                                  CreateIncorrectDimensionsErrorMsg(4,
-                                                                    output.GetNumDimensions(),
-                                                                    batchToSpaceNdLayerStr,
-                                                                    outputTensorStr).data());
-
-    supported &= CheckSupportRule(TensorNumDimensionsAreCorrect(input, 4),
-                                  reasonIfUnsupported,
-                                  CreateIncorrectDimensionsErrorMsg(4,
-                                                                    input.GetNumDimensions(),
-                                                                    batchToSpaceNdLayerStr,
-                                                                    inputTensorStr).data());
-
     return supported;
 }