37 #include "tests/benchmark/fixtures/AlexNetFixture.h" 50 #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC 64 NEFullyConnectedLayer,
73 framework::dataset::
combine(alex_net_data_types,
74 framework::dataset::
make("Batches", { 1, 2, 4 })));
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.
AlexNetFixture< ITensor, Tensor, SubTensor, Accessor, NEActivationLayer, NEConvolutionLayer, NEConvolutionLayer, NEFullyConnectedLayer, NENormalizationLayer, NEPoolingLayer, NESoftmaxLayer > NEAlexNetFixture
Basic function to compute a SoftmaxLayer.
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.