Searched refs:CreateGraphAndRunTest (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/armnn/src/armnn/test/ |
H A D | ShapeInferenceTests.cpp | 109 void CreateGraphAndRunTest(const std::vector<TensorShape>& inputShapes, in CreateGraphAndRunTest() function 179 …CreateGraphAndRunTest<ActivationLayer>({{ 5, 7, 6, 2 }}, {{ 5, 7, 6, 2 }}, descriptor, "activation… 184 CreateGraphAndRunTest<AdditionLayer>({{ 5, 7, 6, 2 }, { 5, 7, 6, 2 }}, {{ 5, 7, 6, 2 }}, "add"); 193 CreateGraphAndRunTest<ArgMinMaxLayer>({{ 1, 3, 2, 4 }}, {{ 1, 2, 4 }}, descriptor, "argMinMax"); 199 …CreateGraphAndRunTest<BatchNormalizationLayer>({{ 1, 2, 3, 2 }}, {{ 1, 2, 3, 2 }}, descriptor, "ba… 213 …CreateGraphAndRunTest<BatchToSpaceNdLayer>({{ 4, 2, 2, 1 }}, {{ 1, 4, 4, 1 }}, descriptor, "batcht… 220 CreateGraphAndRunTest<ComparisonLayer>({{ 5, 7, 6, 2 }, { 5, 7, 6, 2 }}, 230 CreateGraphAndRunTest<ComparisonLayer>({{ 5, 7, 6, 2 }, { 1 }}, 240 CreateGraphAndRunTest<ComparisonLayer>({{ 1 }, { 5, 7, 6, 2 }}, 253 …CreateGraphAndRunTest<ConcatLayer>({{ 1, 2, 1 }, { 1, 2, 1 }}, {{ 2, 2, 1 }}, descriptor, "concat"… [all …]
|