From f0cefbfcb1567b952860824a4767ed15f1a87594 Mon Sep 17 00:00:00 2001 From: Gian Marco Iodice Date: Tue, 17 Apr 2018 09:54:37 +0100 Subject: COMPMID-959 - Fixing subtensor validation Change-Id: I51c878ec9f19da9d05777271d17ecb740387cac8 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/127934 Tested-by: Jenkins Reviewed-by: Georgios Pinitas --- tests/networks/AlexNetNetwork.h | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) (limited to 'tests/networks/AlexNetNetwork.h') diff --git a/tests/networks/AlexNetNetwork.h b/tests/networks/AlexNetNetwork.h index 97991b0121..fa3734d106 100644 --- a/tests/networks/AlexNetNetwork.h +++ b/tests/networks/AlexNetNetwork.h @@ -90,18 +90,18 @@ public: w11 = std::unique_ptr(new SubTensorType(&w[1], TensorShape(5U, 5U, 48U, 128U), Coordinates())); w12 = std::unique_ptr(new SubTensorType(&w[1], TensorShape(5U, 5U, 48U, 128U), Coordinates(0, 0, 0, 128))); - b11 = std::unique_ptr(new SubTensorType(&b[1], TensorShape(128U), Coordinates())); - b12 = std::unique_ptr(new SubTensorType(&b[1], TensorShape(128U), Coordinates(128))); + b11 = std::unique_ptr(new SubTensorType(&b[1], TensorShape(128U), Coordinates(), true)); + b12 = std::unique_ptr(new SubTensorType(&b[1], TensorShape(128U), Coordinates(128), true)); w31 = std::unique_ptr(new SubTensorType(&w[3], TensorShape(3U, 3U, 192U, 192U), Coordinates())); w32 = std::unique_ptr(new SubTensorType(&w[3], TensorShape(3U, 3U, 192U, 192U), Coordinates(0, 0, 0, 192))); - b31 = std::unique_ptr(new SubTensorType(&b[3], TensorShape(192U), Coordinates())); - b32 = std::unique_ptr(new SubTensorType(&b[3], TensorShape(192U), Coordinates(192))); + b31 = std::unique_ptr(new SubTensorType(&b[3], TensorShape(192U), Coordinates(), true)); + b32 = std::unique_ptr(new SubTensorType(&b[3], TensorShape(192U), Coordinates(192), true)); w41 = std::unique_ptr(new SubTensorType(&w[4], TensorShape(3U, 3U, 192U, 128U), Coordinates())); w42 = std::unique_ptr(new SubTensorType(&w[4], TensorShape(3U, 3U, 192U, 128U), Coordinates(0, 0, 0, 128))); - b41 = std::unique_ptr(new SubTensorType(&b[4], TensorShape(128U), Coordinates())); - b42 = std::unique_ptr(new SubTensorType(&b[4], TensorShape(128U), Coordinates(128))); + b41 = std::unique_ptr(new SubTensorType(&b[4], TensorShape(128U), Coordinates(), true)); + b42 = std::unique_ptr(new SubTensorType(&b[4], TensorShape(128U), Coordinates(128), true)); } else { -- cgit v1.2.1