10 #include <doctest/doctest.h> 13 using namespace armnn;
17 TEST_CASE(
"SingleAxisPriorityListTest")
19 std::vector<MemBlock> memBlocks =
fsrcnn;
21 auto singleAxisPriorityList = std::make_shared<SingleAxisPriorityList>();
23 CHECK_EQ(singleAxisPriorityList->GetName(), std::string(
"SingleAxisPriorityList"));
29 std::vector<MemBin> memBins;
31 CHECK_NOTHROW(memBins = validator.Optimize(memBlocks));
34 size_t actualSize = 0;
35 for (
auto memBin : memBins)
37 actualSize += memBin.m_MemSize;
40 CHECK(minMemSize == actualSize);
TEST_SUITE("TestConstTensorLayerVisitor")
std::vector< armnn::MemBlock > fsrcnn
size_t GetMinPossibleMemorySize(const std::vector< armnn::MemBlock > &blocks)
Copyright (c) 2021 ARM Limited and Contributors.
void SetStrategy(std::shared_ptr< IMemoryOptimizerStrategy > strategy)