ReduceMultipleAxesTests.cpp File Reference

#include "../GraphUtils.hpp"
#include "../TestUtils.hpp"
#include <armnn/INetwork.hpp>
#include <doctest/doctest.h>

Go to the source code of this file.

Functions
	TEST_SUITE ("Optimizer_ReduceMultipleAxesCpu")
	TEST_SUITE ("Optimizer_ReduceMultipleAxesGpu")

Function Documentation

TEST_SUITE() [1/2]

TEST_SUITE

(

"Optimizer_ReduceMultipleAxesCpu"

)

Definition at line 251 of file ReduceMultipleAxesTests.cpp.

References armnn::CpuAcc.

{
TEST_CASE("ReduceSumWithTwoAxesKeepDimsCpuAccTest")
{
    ReduceSumWithTwoAxesKeepDimsTest(Compute::CpuAcc);
}

TEST_CASE("ReduceSumWithTwoAxesCpuAccTest")
{
    ReduceSumWithTwoAxesTest(Compute::CpuAcc);
}

TEST_CASE("ReduceSumWithThreeAxesKeepDimsCpuAccTest")
{
    ReduceSumWithThreeAxesKeepDimsTest(Compute::CpuAcc);
}

TEST_CASE("ReduceSumWithThreeAxesCpuAccTest")
{
    ReduceSumWithThreeAxesTest(Compute::CpuAcc);
}
}

TEST_SUITE() [2/2]

TEST_SUITE

(

"Optimizer_ReduceMultipleAxesGpu"

)

Definition at line 276 of file ReduceMultipleAxesTests.cpp.

References armnn::GpuAcc.

{
TEST_CASE("ReduceSumWithTwoAxesKeepDimsGpuAccTest")
{
    ReduceSumWithTwoAxesKeepDimsTest(Compute::GpuAcc);
}

TEST_CASE("ReduceSumWithTwoAxesGpuAccTest")
{
    ReduceSumWithTwoAxesTest(Compute::GpuAcc);
}

TEST_CASE("ReduceSumWithThreeAxesKeepDimsGpuAccTest")
{
    ReduceSumWithThreeAxesKeepDimsTest(Compute::GpuAcc);
}

TEST_CASE("ReduceSumWithThreeAxesGpuAccTest")
{
    ReduceSumWithThreeAxesTest(Compute::GpuAcc);
}
}