From 3112e33d8d0b987e85107390a0350bd5988f5f01 Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Wed, 21 Nov 2018 15:44:55 +0000 Subject: COMPMID-1451 Change PriorBox output to NCHw Output of Priorbox should be independent of the input data layout and should always be in NCHW format Change-Id: Ie80cd4e51c78945b158c0db1af1923bdf8d7ea7b --- tests/validation/fixtures/PriorBoxLayerFixture.h | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) (limited to 'tests/validation/fixtures') diff --git a/tests/validation/fixtures/PriorBoxLayerFixture.h b/tests/validation/fixtures/PriorBoxLayerFixture.h index dd7a49ee1f..fb15631789 100644 --- a/tests/validation/fixtures/PriorBoxLayerFixture.h +++ b/tests/validation/fixtures/PriorBoxLayerFixture.h @@ -62,13 +62,12 @@ protected: if(data_layout == DataLayout::NHWC) { permute(input_shape, PermutationVector(2U, 0U, 1U)); - permute(output_shape, PermutationVector(2U, 0U, 1U)); } // Create tensors TensorType src1 = create_tensor(input_shape, data_type, 1, QuantizationInfo(), data_layout); TensorType src2 = create_tensor(input_shape, data_type, 1, QuantizationInfo(), data_layout); - TensorType dst = create_tensor(output_shape, data_type, 1, QuantizationInfo(), data_layout); + TensorType dst = create_tensor(output_shape, data_type, 1, QuantizationInfo()); // Create and configure function FunctionType prior_box; -- cgit v1.2.1