aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/NEON/FullyConnectedLayer.cpp
diff options
context:
space:
mode:
authorMoritz Pflanzer <moritz.pflanzer@arm.com>2017-06-30 12:48:43 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-09-17 14:15:39 +0100
commit94450f1fc91a89778354c2e1c07a328ba86d9cfc (patch)
tree9a84b1b86f5361027492b6698ea946d2e321baba /tests/validation/NEON/FullyConnectedLayer.cpp
parentd4f8c27f08c968f608a9e0e9b88df1871f62cdd3 (diff)
downloadComputeLibrary-94450f1fc91a89778354c2e1c07a328ba86d9cfc.tar.gz
COMPMID-417: Use a common create_tensor function
Change-Id: I6b0511484a5b433ebec3fd62d778e64dcb4f89b5 Reviewed-on: http://mpd-gerrit.cambridge.arm.com/79362 Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com> Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Diffstat (limited to 'tests/validation/NEON/FullyConnectedLayer.cpp')
-rw-r--r--tests/validation/NEON/FullyConnectedLayer.cpp19
1 files changed, 10 insertions, 9 deletions
diff --git a/tests/validation/NEON/FullyConnectedLayer.cpp b/tests/validation/NEON/FullyConnectedLayer.cpp
index bda235bd55..69ff663193 100644
--- a/tests/validation/NEON/FullyConnectedLayer.cpp
+++ b/tests/validation/NEON/FullyConnectedLayer.cpp
@@ -21,10 +21,11 @@
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
-#include "NEON/Helper.h"
#include "NEON/NEAccessor.h"
#include "TypePrinter.h"
#include "dataset/FullyConnectedLayerDataset.h"
+#include "tests/Globals.h"
+#include "tests/Utils.h"
#include "validation/Datasets.h"
#include "validation/Reference.h"
#include "validation/Validation.h"
@@ -49,9 +50,9 @@ Tensor compute_fully_connected_layer(const TensorShape &input_shape, const Tenso
bool transpose_weights, int fixed_point_position)
{
// Create tensors
- Tensor src = create_tensor(input_shape, dt, 1, fixed_point_position);
- Tensor bias = create_tensor(bias_shape, dt, 1, fixed_point_position);
- Tensor dst = create_tensor(output_shape, dt, 1, fixed_point_position);
+ Tensor src = create_tensor<Tensor>(input_shape, dt, 1, fixed_point_position);
+ Tensor bias = create_tensor<Tensor>(bias_shape, dt, 1, fixed_point_position);
+ Tensor dst = create_tensor<Tensor>(output_shape, dt, 1, fixed_point_position);
// Swap the first and second dimension of weights' shape if transpose_weights is true
TensorShape ws = weights_shape;
@@ -62,7 +63,7 @@ Tensor compute_fully_connected_layer(const TensorShape &input_shape, const Tenso
ws.set(1, dimx);
}
- Tensor weights = create_tensor(ws, dt, 1, fixed_point_position);
+ Tensor weights = create_tensor<Tensor>(ws, dt, 1, fixed_point_position);
// Create and configure function.
// Note: We pass the weights already transposed
@@ -115,9 +116,9 @@ BOOST_DATA_TEST_CASE(Configuration,
int fixed_point_position = (dt == DataType::F32) ? 0 : 3;
// Create tensors
- Tensor src = create_tensor(fc_set.src_shape, dt, 1, fixed_point_position);
- Tensor bias = create_tensor(fc_set.bias_shape, dt, 1, fixed_point_position);
- Tensor dst = create_tensor(fc_set.dst_shape, dt, 1, fixed_point_position);
+ Tensor src = create_tensor<Tensor>(fc_set.src_shape, dt, 1, fixed_point_position);
+ Tensor bias = create_tensor<Tensor>(fc_set.bias_shape, dt, 1, fixed_point_position);
+ Tensor dst = create_tensor<Tensor>(fc_set.dst_shape, dt, 1, fixed_point_position);
// Swap the first and second dimension of weights' shape if transpose_weights is true
TensorShape ws = fc_set.weights_shape;
@@ -128,7 +129,7 @@ BOOST_DATA_TEST_CASE(Configuration,
ws.set(1, dimx);
}
- Tensor weights = create_tensor(ws, dt, 1, fixed_point_position);
+ Tensor weights = create_tensor<Tensor>(ws, dt, 1, fixed_point_position);
BOOST_TEST(src.info()->is_resizable());
BOOST_TEST(weights.info()->is_resizable());