Searched refs:testDynamicOutputShape (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/runtime/test/ |
| D | GeneratedTestUtils.cpp | 35 static OperandType getOperandType(const TestOperand& op, bool testDynamicOutputShape) { in getOperandType() argument
37 if (testDynamicOutputShape && op.lifetime == TestOperandLifeTime::SUBGRAPH_OUTPUT) { in getOperandType()
105 static void createModelFromSubgraph(const TestSubgraph& subgraph, bool testDynamicOutputShape, in createModelFromSubgraph() argument
111 auto type = getOperandType(operand, testDynamicOutputShape); in createModelFromSubgraph()
134 createModelFromSubgraph(refSubgraph, testDynamicOutputShape, refSubgraphs, in createModelFromSubgraph()
158 void createModel(const TestModel& testModel, bool testDynamicOutputShape, GeneratedModel* model) { in createModel() argument
164 createModelFromSubgraph(testModel.main, testDynamicOutputShape, testModel.referenced, memory, in createModel()
|
| D | GeneratedTestUtils.h | 71 void createModel(const test_helper::TestModel& testModel, bool testDynamicOutputShape,
|
| /frameworks/ml/nn/tools/test_generator/ |
| D | test_generator.py | 1098 self.testDynamicOutputShape = True
1297 self.testDynamicOutputShape = False
|