diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2018-11-21 15:44:55 +0000 |
---|---|---|
committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2018-11-21 17:32:46 +0000 |
commit | 3112e33d8d0b987e85107390a0350bd5988f5f01 (patch) | |
tree | 050c8400608caaa216305611ca2de1d8163e462b /tests/validation | |
parent | edf26ea4921fd0c162abb962fc4863846e73f53f (diff) | |
download | ComputeLibrary-3112e33d8d0b987e85107390a0350bd5988f5f01.tar.gz |
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
Diffstat (limited to 'tests/validation')
-rw-r--r-- | tests/validation/fixtures/PriorBoxLayerFixture.h | 3 |
1 files changed, 1 insertions, 2 deletions
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<TensorType>(input_shape, data_type, 1, QuantizationInfo(), data_layout); TensorType src2 = create_tensor<TensorType>(input_shape, data_type, 1, QuantizationInfo(), data_layout); - TensorType dst = create_tensor<TensorType>(output_shape, data_type, 1, QuantizationInfo(), data_layout); + TensorType dst = create_tensor<TensorType>(output_shape, data_type, 1, QuantizationInfo()); // Create and configure function FunctionType prior_box; |