#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 251 of file ReduceMultipleAxesTests.cpp.
References armnn::CpuAcc.
253 TEST_CASE(
"ReduceSumWithTwoAxesKeepDimsCpuAccTest")
255 ReduceSumWithTwoAxesKeepDimsTest(Compute::CpuAcc);
258 TEST_CASE(
"ReduceSumWithTwoAxesCpuAccTest")
260 ReduceSumWithTwoAxesTest(Compute::CpuAcc);
263 TEST_CASE(
"ReduceSumWithThreeAxesKeepDimsCpuAccTest")
265 ReduceSumWithThreeAxesKeepDimsTest(Compute::CpuAcc);
268 TEST_CASE(
"ReduceSumWithThreeAxesCpuAccTest")
270 ReduceSumWithThreeAxesTest(Compute::CpuAcc);
◆ TEST_SUITE() [2/2]
TEST_SUITE |
( |
"Optimizer_ReduceMultipleAxesGpu" |
| ) |
|
Definition at line 276 of file ReduceMultipleAxesTests.cpp.
References armnn::GpuAcc.
278 TEST_CASE(
"ReduceSumWithTwoAxesKeepDimsGpuAccTest")
280 ReduceSumWithTwoAxesKeepDimsTest(Compute::GpuAcc);
283 TEST_CASE(
"ReduceSumWithTwoAxesGpuAccTest")
285 ReduceSumWithTwoAxesTest(Compute::GpuAcc);
288 TEST_CASE(
"ReduceSumWithThreeAxesKeepDimsGpuAccTest")
290 ReduceSumWithThreeAxesKeepDimsTest(Compute::GpuAcc);
293 TEST_CASE(
"ReduceSumWithThreeAxesGpuAccTest")
295 ReduceSumWithThreeAxesTest(Compute::GpuAcc);