30 #include "tests/benchmark/fixtures/DirectConvolutionLayerFixture.h" 31 #include "tests/datasets/DirectConvolutionLayerDataset.h" 32 #include "tests/datasets/system_tests/alexnet/AlexNetConvolutionLayerDataset.h" 33 #include "tests/datasets/system_tests/googlenet/inceptionv1/GoogLeNetInceptionV1ConvolutionLayerDataset.h" 34 #include "tests/datasets/system_tests/googlenet/inceptionv4/GoogLeNetInceptionV4ConvolutionLayerDataset.h" 35 #include "tests/datasets/system_tests/squeezenet/SqueezeNetConvolutionLayerDataset.h" 36 #include "tests/datasets/system_tests/vgg/vgg16/VGG16ConvolutionLayerDataset.h" 37 #include "tests/datasets/system_tests/yolo/v2/YOLOV2ConvolutionLayerDataset.h" 51 #ifdef ARM_COMPUTE_ENABLE_F16 63 framework::dataset::
combine(framework::dataset::
combine(framework::dataset::
combine(datasets::AlexNetDirectConvolutionLayerDataset(),
66 framework::dataset::
make("Batches", 1)));
69 framework::dataset::
combine(framework::dataset::
combine(framework::dataset::
combine(datasets::GoogLeNetInceptionV1DirectConvolutionLayerDataset(),
72 framework::dataset::
make("Batches", 1)));
75 framework::dataset::
combine(framework::dataset::
combine(framework::dataset::
combine(datasets::GoogLeNetInceptionV4DirectConvolutionLayerDataset(),
78 framework::dataset::
make("Batches", 1)));
81 framework::dataset::
combine(framework::dataset::
combine(framework::dataset::
combine(datasets::SqueezeNetConvolutionLayerDataset(),
82 framework::dataset::
make("ActivationInfo", ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
84 framework::dataset::
make("Batches", 1)));
88 framework::dataset::
combine(framework::dataset::
combine(framework::dataset::
combine(datasets::AlexNetDirectConvolutionLayerDataset(),
89 framework::dataset::
make("ActivationInfo", ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
91 framework::dataset::
make("Batches", { 4, 8 })));
1 channel, 1 F32 per channel
std::enable_if< is_container< T >::value, ContainerDataset< T > >::type make(std::string name, T &&values)
Helper function to create a ContainerDataset.
Activation Layer Information class.
This file contains all available output stages for GEMMLowp on OpenCL.
1 channel, 1 F16 per channel
#define TEST_SUITE(SUITE_NAME)
DatasetMode
Possible dataset modes.
REGISTER_FIXTURE_DATA_TEST_CASE(MobileNetBatchNormalizationLayer, CLBatchNormalizationLayerFixture, framework::DatasetMode::ALL, combine(combine(combine(combine(combine(datasets::MobileNetBatchNormalizationLayerDataset(), combine(framework::dataset::make("UseGamma",{false, true}), framework::dataset::make("UseBeta",{false, true}))), framework::dataset::make("ActivationInfo", ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::BOUNDED_RELU, 6.f))), data_types), framework::dataset::make("DataLayout",{DataLayout::NCHW, DataLayout::NHWC})), framework::dataset::make("Batches", 1)))
CartesianProductDataset< T, U > combine(T &&dataset1, U &&dataset2)
Helper function to create a CartesianProductDataset.
DirectConvolutionLayerFixture< Tensor, NEDirectConvolutionLayer, Accessor > NEDirectConvolutionLayerFixture