#include "../GraphUtils.hpp"
#include "../TestUtils.hpp"
#include <armnn/INetwork.hpp>
#include <doctest/doctest.h>
Go to the source code of this file.
◆ TEST_SUITE() [1/2]
TEST_SUITE |
( |
"Optimizer_ReduceMultipleAxesCpu" |
| ) |
|
Definition at line 253 of file ReduceMultipleAxesTests.cpp.
References armnn::CpuAcc.
255 TEST_CASE(
"ReduceSumWithTwoAxesKeepDimsCpuAccTest")
257 ReduceSumWithTwoAxesKeepDimsTest(Compute::CpuAcc);
260 TEST_CASE(
"ReduceSumWithTwoAxesCpuAccTest")
262 ReduceSumWithTwoAxesTest(Compute::CpuAcc);
265 TEST_CASE(
"ReduceSumWithThreeAxesKeepDimsCpuAccTest")
267 ReduceSumWithThreeAxesKeepDimsTest(Compute::CpuAcc);
270 TEST_CASE(
"ReduceSumWithThreeAxesCpuAccTest")
272 ReduceSumWithThreeAxesTest(Compute::CpuAcc);
◆ TEST_SUITE() [2/2]
TEST_SUITE |
( |
"Optimizer_ReduceMultipleAxesGpu" |
| ) |
|
Definition at line 278 of file ReduceMultipleAxesTests.cpp.
References armnn::GpuAcc.
280 TEST_CASE(
"ReduceSumWithTwoAxesKeepDimsGpuAccTest")
282 ReduceSumWithTwoAxesKeepDimsTest(Compute::GpuAcc);
285 TEST_CASE(
"ReduceSumWithTwoAxesGpuAccTest")
287 ReduceSumWithTwoAxesTest(Compute::GpuAcc);
290 TEST_CASE(
"ReduceSumWithThreeAxesKeepDimsGpuAccTest")
292 ReduceSumWithThreeAxesKeepDimsTest(Compute::GpuAcc);
295 TEST_CASE(
"ReduceSumWithThreeAxesGpuAccTest")
297 ReduceSumWithThreeAxesTest(Compute::GpuAcc);