Lines Matching refs:WrapperModel
60 using WrapperModel = nn::test_wrapper::Model; typedef
203 WrapperModel mModel;
206 void createSimpleAddModel(WrapperModel* model) { in createSimpleAddModel()
1140 void createAddMaxModel(WrapperModel* model, bool reverseOrder) { in createAddMaxModel()
1205 void createAddMulModel(WrapperModel* model, bool reverseOrder) { in createAddMulModel()
1251 void createCondModel(WrapperModel* model, bool dynamicRank) { in createCondModel()
1265 void addReluOperation(WrapperModel* model, std::vector<uint32_t>* modelInputIndexes, in addReluOperation()
1279 void createReluModel(WrapperModel* model, bool dynamicRank) { in createReluModel()
1286 void addWhileOperation(std::vector<WrapperModel>* extraModels, WrapperModel* mainModel, in addWhileOperation()
1295 WrapperModel* condModel = &extraModels->at(extraModels->size() - 2); in addWhileOperation()
1296 WrapperModel* bodyModel = &extraModels->at(extraModels->size() - 1); in addWhileOperation()
1316 void createReluStaticWhileModel(std::vector<WrapperModel>* extraModels, WrapperModel* mainModel) { in createReluStaticWhileModel()
1343 std::vector<WrapperModel> extraModels; in TEST_F()
1348 mModel = WrapperModel{}; in TEST_F()
1364 std::vector<WrapperModel> extraModels; in TEST_F()
1369 mModel = WrapperModel{}; in TEST_F()
1372 void createStaticWhileDynamicWhileModel(std::vector<WrapperModel>* extraModels, in createStaticWhileDynamicWhileModel()
1373 WrapperModel* mainModel) { in createStaticWhileDynamicWhileModel()
1401 std::vector<WrapperModel> extraModels; in TEST_F()
1406 mModel = WrapperModel{}; in TEST_F()