Compute Library
18.05
|
#include "arm_compute/core/TensorShape.h"
#include "arm_compute/core/Types.h"
#include "arm_compute/runtime/CL/CLTensor.h"
#include "arm_compute/runtime/CL/CLTensorAllocator.h"
#include "arm_compute/runtime/CL/functions/CLSoftmaxLayer.h"
#include "tests/CL/CLAccessor.h"
#include "tests/benchmark/fixtures/SoftmaxLayerFixture.h"
#include "tests/datasets/ShapeDatasets.h"
#include "tests/framework/Macros.h"
#include "tests/framework/datasets/Datasets.h"
#include "utils/TypePrinter.h"
Go to the source code of this file.
Namespaces | |
arm_compute | |
This file contains all available output stages for GEMMLowp on OpenCL. | |
arm_compute::test | |
arm_compute::test::benchmark | |
Typedefs | |
using | CLSoftmaxLayerFixture = SoftmaxLayerFixture< CLTensor, CLSoftmaxLayer, CLAccessor > |
Functions | |
REGISTER_FIXTURE_DATA_TEST_CASE (SoftmaxLayerSmall, CLSoftmaxLayerFixture, framework::DatasetMode::ALL, framework::dataset::combine(datasets::SoftmaxLayerSmallShapes(), data_types)) | |
REGISTER_FIXTURE_DATA_TEST_CASE (SoftmaxLayerLarge, CLSoftmaxLayerFixture, framework::DatasetMode::NIGHTLY, framework::dataset::combine(datasets::SoftmaxLayerLargeShapes(), data_types)) | |