Compute Library
 22.05
Unstack.cpp File Reference
#include "arm_compute/core/Types.h"
#include "arm_compute/runtime/NEON/functions/NEUnstack.h"
#include "arm_compute/runtime/Tensor.h"
#include "arm_compute/runtime/TensorAllocator.h"
#include "tests/NEON/Accessor.h"
#include "tests/datasets/ShapeDatasets.h"
#include "tests/framework/Asserts.h"
#include "tests/framework/Macros.h"
#include "tests/framework/datasets/Datasets.h"
#include "tests/validation/Validation.h"
#include "tests/validation/fixtures/UnstackFixture.h"

Go to the source code of this file.

Namespaces

 arm_compute
 Copyright (c) 2017-2022 Arm Limited.
 
 arm_compute::test
 
 arm_compute::test::validation
 

Typedefs

template<typename T >
using NEUnstackFixture = UnstackValidationFixture< Tensor, ITensor, Accessor, NEUnstack, T >
 

Functions

 DATA_TEST_CASE (Validate, framework::DatasetMode::ALL, zip(zip(zip(zip(framework::dataset::make("InputInfo", { TensorInfo(TensorShape(1U, 9U, 8U), 1, DataType::U8), TensorInfo(TensorShape(1U, 2U, 3U), 1, DataType::U8), TensorInfo(TensorShape(1U, 2U, 3U), 1, DataType::S32), TensorInfo(TensorShape(3U, 7U, 5U), 1, DataType::S32), TensorInfo(TensorShape(13U, 7U, 5U), 1, DataType::S16), TensorInfo(TensorShape(1U, 2U, 3U), 1, DataType::U8), }), framework::dataset::make("OutputInfo", { std::vector< TensorInfo >{ TensorInfo(TensorShape(9U, 8U), 1, DataType::U8) }, std::vector< TensorInfo >{ TensorInfo(TensorShape(2U, 3U), 1, DataType::U8) }, std::vector< TensorInfo >{ TensorInfo(TensorShape(2U, 3U), 1, DataType::S32) }, std::vector< TensorInfo >{ TensorInfo(TensorShape(7U, 5U), 1, DataType::S32), TensorInfo(TensorShape(7U, 5U), 1, DataType::S32), TensorInfo(TensorShape(7U, 5U), 1, DataType::S32) }, std::vector< TensorInfo >{ TensorInfo(TensorShape(7U, 5U), 1, DataType::S16) }, std::vector< TensorInfo >{ TensorInfo(TensorShape(9U, 8U), 1, DataType::S32) }, })), framework::dataset::make("Axis", { -3, 3, -4, -3, 1, 1 })), framework::dataset::make("Num", { 1, 1, 1, 1, 0, 1 })), framework::dataset::make("Expected", { true, false, false, true, false, false })), input_info, output_info, axis, num, expected)
 
 for (size_t k=0;k< _target.size();++k)
 
 TEST_SUITE_END () FIXTURE_DATA_TEST_CASE(RunSmall = ActivationValidationQuantizedFixture<CLTensor, CLAccessor, CLActivationLayer, T>
 [CLActivationLayer Test snippet] More...