24.04
|
#include "arm_compute/core/Types.h"
#include "arm_compute/dynamic_fusion/sketch/gpu/operators/GpuSoftmax.h"
#include "tests/CL/CLAccessor.h"
#include "tests/datasets/ShapeDatasets.h"
#include "tests/framework/Asserts.h"
#include "tests/framework/datasets/Datasets.h"
#include "tests/framework/Fixture.h"
#include "tests/framework/Macros.h"
#include "tests/validation/fixtures/dynamic_fusion/operators/SoftmaxFixture.h"
#include "tests/validation/Validation.h"
Go to the source code of this file.
Namespaces | |
arm_compute | |
Copyright (c) 2017-2024 Arm Limited. | |
arm_compute::test | |
arm_compute::test::validation | |
Typedefs | |
template<typename T > | |
using | DynamicFusionSoftmaxLayerFixture = DynamicFusionSoftmaxValidationFixture< CLTensor, CLAccessor, GpuSoftmax, T > |
Functions | |
RelativeTolerance< half > | tolerance_f16 (half(0.2)) |
Tolerance for float operations. More... | |
constexpr AbsoluteTolerance< float > | tolerance_f32 (0.001f) |
Tolerance value for comparing reference's output against implementation's output for 32-bit floating-point type. More... | |
DATA_TEST_CASE (Validate, framework::DatasetMode::DISABLED, zip(make("InputInfo", { TensorInfo(TensorShape(27U, 13U), 1, DataType::F32), TensorInfo(TensorShape(27U, 13U), 1, DataType::F32), TensorInfo(TensorShape(32U, 13U), 1, DataType::F32), TensorInfo(TensorShape(32U, 13U), 1, DataType::F32), TensorInfo(TensorShape(32U, 13U), 1, DataType::S32), TensorInfo(TensorShape(32U, 13U), 1, DataType::F16), TensorInfo(TensorShape(32U, 13U), 1, DataType::F32), TensorInfo(TensorShape(32U, 13U), 1, DataType::F32), TensorInfo(TensorShape(32U, 13U), 1, DataType::F32), TensorInfo(TensorShape(32U, 13U), 1, DataType::F32), }), make("OutputInfo",{ TensorInfo(TensorShape(27U, 13U), 1, DataType::F16), TensorInfo(TensorShape(27U, 11U), 1, DataType::F32), TensorInfo(TensorShape(32U, 13U), 1, DataType::F32), TensorInfo(TensorShape(32U, 13U), 1, DataType::F32), TensorInfo(TensorShape(32U, 13U), 1, DataType::F32), TensorInfo(TensorShape(32U, 13U), 1, DataType::QASYMM16), TensorInfo(TensorShape(32U, 13U), 1, DataType::F32), TensorInfo(TensorShape(32U, 13U), 1, DataType::F32), TensorInfo(TensorShape(32U, 13U), 1, DataType::F32), TensorInfo(TensorShape(32U, 13U), 1, DataType::F32), }), make("beta", { 1.0, 2.0, 2.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, }), make("axis", { 0, 0, 1, 0, 0, 0, -3, 2, 1, -1, }), make("Expected", { false, false, false, true, false, false, false, false, false, false})), input_info, output_info, beta, axis, expected) | |
TODO: COMPMID-6713 Softmax is not implemented in CKW. More... | |
FIXTURE_DATA_TEST_CASE (RunSmall, DynamicFusionSoftmaxLayerFixture< float >, framework::DatasetMode::DISABLED, combine(datasets::SoftmaxLayerSmallShapes(), make("DataType", DataType::F32), make("Beta", { 1.0f, 2.0f }), make("Axis", { 0 }), make("is_log", {false, true}))) | |
FIXTURE_DATA_TEST_CASE (RunLarge, DynamicFusionSoftmaxLayerFixture< float >, framework::DatasetMode::DISABLED, combine(datasets::SoftmaxLayerLargeShapes(), make("DataType", DataType::F32), make("Beta", { 1.0f, 2.0f }), make("Axis", { 0 }), make("is_log", {false, true}))) | |
FIXTURE_DATA_TEST_CASE (Run4D, DynamicFusionSoftmaxLayerFixture< float >, framework::DatasetMode::DISABLED, combine(datasets::SoftmaxLayer4DShapes(), make("DataType", DataType::F32), make("Beta", { 1.0f, 2.0f }), make("Axis", { 0 }), make("is_log", {false, true}))) | |
TEST_SUITE_END () FIXTURE_DATA_TEST_CASE(RunSmall = ActivationValidationQuantizedFixture<CLTensor, CLAccessor, CLActivationLayer, T> | |
[CLActivationLayer Test snippet] More... | |
combine (datasets::SoftmaxLayerSmallShapes(), make("DataType", DataType::F16), make("Beta", { 1.0f, 2.0f }), make("Axis", { 0 }), make("is_log", {false, true}))) | |
FIXTURE_DATA_TEST_CASE (RunLarge, DynamicFusionSoftmaxLayerFixture< half >, framework::DatasetMode::DISABLED, combine(datasets::SoftmaxLayerLargeShapes(), make("DataType", DataType::F16), make("Beta", { 1.0f, 2.0f }), make("Axis", { 0 }), make("is_log", {false, true}))) | |
FIXTURE_DATA_TEST_CASE (Run4D, DynamicFusionSoftmaxLayerFixture< half >, framework::DatasetMode::DISABLED, combine(datasets::SoftmaxLayer4DShapes(), make("DataType", DataType::F16), make("Beta", { 1.0f, 2.0f }), make("Axis", { 0 }), make("is_log", {false, true}))) | |
Variables | |
DynamicFusionSoftmaxLayerFixture< half > | |