14 #include <doctest/doctest.h> 18 TEST_CASE(
"CopyBetweenCpuAndGpu")
21 MemCopyTest<armnn::RefWorkloadFactory, armnn::ClWorkloadFactory, armnn::DataType::Float32>(
false);
24 CHECK_MESSAGE(predResult.m_Result, predResult.m_Message.str());
27 TEST_CASE(
"CopyBetweenGpuAndCpu")
30 MemCopyTest<armnn::ClWorkloadFactory, armnn::RefWorkloadFactory, armnn::DataType::Float32>(
false);
33 CHECK_MESSAGE(predResult.m_Result, predResult.m_Message.str());
36 TEST_CASE(
"CopyBetweenCpuAndGpuWithSubtensors")
39 MemCopyTest<armnn::RefWorkloadFactory, armnn::ClWorkloadFactory, armnn::DataType::Float32>(
true);
42 CHECK_MESSAGE(predResult.m_Result, predResult.m_Message.str());
45 TEST_CASE(
"CopyBetweenGpuAndCpuWithSubtensors")
48 MemCopyTest<armnn::ClWorkloadFactory, armnn::RefWorkloadFactory, armnn::DataType::Float32>(
true);
51 CHECK_MESSAGE(predResult.m_Result, predResult.m_Message.str());
armnn::TensorShape m_ExpectedShape
armnn::PredicateResult CompareTensors(const std::vector< T > &actualData, const std::vector< T > &expectedData, const armnn::TensorShape &actualShape, const armnn::TensorShape &expectedShape, bool compareBoolean=false, bool isDynamic=false)
std::vector< T > m_ExpectedData
std::vector< T > m_ActualData
armnn::TensorShape m_ActualShape