Home
last modified time | relevance | path

Searched refs:TestSubgraph (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/tools/test_generator/test_harness/include/
DTestHarness.h292 struct TestSubgraph { struct
300 TestSubgraph main; argument
301 std::vector<TestSubgraph> referenced;
318 void forEachSubgraph(std::function<void(const TestSubgraph&)> handler) const { in forEachSubgraph()
320 for (const TestSubgraph& subgraph : referenced) { in forEachSubgraph()
325 void forEachSubgraph(std::function<void(TestSubgraph&)> handler) { in forEachSubgraph()
327 for (TestSubgraph& subgraph : referenced) { in forEachSubgraph()
335 newTestModel.forEachSubgraph([](TestSubgraph& subgraph) { in copy()
345 forEachSubgraph([&result](const TestSubgraph& subgraph) { in hasQuant8CoupledOperands()
394 forEachSubgraph([&result](const TestSubgraph& subgraph) { in hasScalarOutputs()
/frameworks/ml/nn/runtime/test/
DGeneratedTestUtils.cpp90 auto processSubgraph = [&size](const TestSubgraph& subgraph) { in createConstantReferenceMemory()
99 for (const TestSubgraph& subgraph : testModel.referenced) { in createConstantReferenceMemory()
105 static void createModelFromSubgraph(const TestSubgraph& subgraph, bool testDynamicOutputShape, in createModelFromSubgraph()
106 const std::vector<TestSubgraph>& refSubgraphs, in createModelFromSubgraph()
131 const TestSubgraph& refSubgraph = refSubgraphs[refIndex]; in createModelFromSubgraph()
/frameworks/ml/nn/tools/test_generator/test_harness/
DTestHarness.cpp276 model.forEachSubgraph([&hasFloat16Inputs](const TestSubgraph& subgraph) { in checkResults()
296 auto processSubgraph = [](TestSubgraph* subgraph) { in convertQuant8AsymmOperandsToSigned()
312 for (TestSubgraph& subgraph : converted.referenced) { in convertQuant8AsymmOperandsToSigned()