MCPcopy Create free account
hub / github.com/ARM-software/armnn / ReplaceUnsupportedLayers

Function ReplaceUnsupportedLayers

src/armnn/test/SubgraphViewTests.cpp:1920–1941  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1918}
1919
1920void ReplaceUnsupportedLayers(SubgraphView& subgraph)
1921{
1922 using ReplacementFunc = bool (*)(SubgraphView&, IConnectableLayer*);
1923 const ReplacementFunc replacementFuncs[] = {
1924 &ReplaceTestMultiplication,
1925 };
1926
1927 subgraph.ForEachLayer([replacementFuncs, &subgraph](IConnectableLayer* layer)
1928 {
1929 auto madeChange = false;
1930 for (const ReplacementFunc f : replacementFuncs)
1931 {
1932 madeChange = f(subgraph, layer);
1933 if (madeChange)
1934 {
1935 goto nextIteration;
1936 }
1937 }
1938 nextIteration:;
1939 }
1940 );
1941}
1942
1943TEST_CASE("SubgraphViewWorkingCopyReplacementFunc")
1944{

Callers 1

Calls 1

ForEachLayerMethod · 0.80

Tested by

no test coverage detected