Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/armnn/test/
H A DShapeInferenceTests.cpp50 void RunShapeInferenceTest(LayerT* const layer, in RunShapeInferenceTest() function
117 RunShapeInferenceTest<LayerT>(layer, dimensionSizeLists); in CreateGraphAndRunTest()
378 …RunShapeInferenceTest<DetectionPostProcessLayer>(layer, {{ 1, 3, 4 }, { 1, 3 }, { 1, 3 }, { 1 }});
471 RunShapeInferenceTest<LstmLayer>(layer, {{2, 80}, {2, 20}, {2, 20}, {2, 20}});
559 RunShapeInferenceTest<QLstmLayer>(layer, {{2, 20}, {2, 20}, {2, 20}});
587 RunShapeInferenceTest<QuantizedLstmLayer>(layer, {{2, 20}, {2, 20}, {2, 20}});