Compute Library
18.05
|
#include "arm_compute/core/TensorShape.h"
#include "arm_compute/core/Types.h"
#include "arm_compute/runtime/CL/CLArray.h"
#include "arm_compute/runtime/CL/CLTensor.h"
#include "arm_compute/runtime/CL/CLTensorAllocator.h"
#include "arm_compute/runtime/CL/functions/CLROIPoolingLayer.h"
#include "tests/CL/CLAccessor.h"
#include "tests/CL/CLArrayAccessor.h"
#include "tests/benchmark/fixtures/ROIPoolingLayerFixture.h"
#include "tests/datasets/ROIPoolingLayerDataset.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 | CLROIPoolingLayerFixture = ROIPoolingLayerFixture< CLTensor, CLROIPoolingLayer, CLAccessor, CLArray< ROI >, CLArrayAccessor< ROI >> |
Functions | |
REGISTER_FIXTURE_DATA_TEST_CASE (SmallROIPoolingLayer, CLROIPoolingLayerFixture, framework::DatasetMode::ALL, framework::dataset::combine(framework::dataset::combine(datasets::SmallROIPoolingLayerDataset(), framework::dataset::make("DataType",{DataType::F16, DataType::F32})), framework::dataset::make("Batches",{1, 4, 8}))) | |