From f77cab57b3eca1425384d4d5bfe44d76fc7023b9 Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Thu, 1 Jun 2023 16:15:13 +0100 Subject: 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 Change-Id: I7bdd11e4936a27cd97ec65fd915e6ccaa1494cff --- src/backends/reference/RefLayerSupport.cpp | 14 -------------- 1 file changed, 14 deletions(-) (limited to 'src/backends/reference/RefLayerSupport.cpp') 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; } -- cgit v1.2.1