Lines Matching refs:WrapperModel
61 using WrapperModel = nn::test_wrapper::Model; typedef
204 WrapperModel mModel;
207 void createSimpleAddModel(WrapperModel* model) { in createSimpleAddModel()
1141 void createAddMaxModel(WrapperModel* model, bool reverseOrder) { in createAddMaxModel()
1206 void createAddMulModel(WrapperModel* model, bool reverseOrder) { in createAddMulModel()
1252 void createCondModel(WrapperModel* model, bool dynamicRank) { in createCondModel()
1266 void addReluOperation(WrapperModel* model, std::vector<uint32_t>* modelInputIndexes, in addReluOperation()
1280 void createReluModel(WrapperModel* model, bool dynamicRank) { in createReluModel()
1287 void addWhileOperation(std::vector<WrapperModel>* extraModels, WrapperModel* mainModel, in addWhileOperation()
1296 WrapperModel* condModel = &extraModels->at(extraModels->size() - 2); in addWhileOperation()
1297 WrapperModel* bodyModel = &extraModels->at(extraModels->size() - 1); in addWhileOperation()
1317 void createReluStaticWhileModel(std::vector<WrapperModel>* extraModels, WrapperModel* mainModel) { in createReluStaticWhileModel()
1344 std::vector<WrapperModel> extraModels; in TEST_F()
1349 mModel = WrapperModel{}; in TEST_F()
1365 std::vector<WrapperModel> extraModels; in TEST_F()
1370 mModel = WrapperModel{}; in TEST_F()
1373 void createStaticWhileDynamicWhileModel(std::vector<WrapperModel>* extraModels, in createStaticWhileDynamicWhileModel()
1374 WrapperModel* mainModel) { in createStaticWhileDynamicWhileModel()
1402 std::vector<WrapperModel> extraModels; in TEST_F()
1407 mModel = WrapperModel{}; in TEST_F()