Lines Matching refs:mainModel
1286 void addWhileOperation(std::vector<WrapperModel>* extraModels, WrapperModel* mainModel, in addWhileOperation() argument
1303 const uint32_t op1 = mainModel->addOperand(&modelType); in addWhileOperation()
1304 const uint32_t op2 = mainModel->addOperand(&modelType); in addWhileOperation()
1305 const uint32_t op3 = mainModel->addOperand(&floatType); in addWhileOperation()
1306 const uint32_t op4 = mainModel->addOperand(&floatType); in addWhileOperation()
1307 mainModel->setOperandValueFromModel(op1, condModel); in addWhileOperation()
1308 mainModel->setOperandValueFromModel(op2, bodyModel); in addWhileOperation()
1310 mainModel->addOperation(ANEURALNETWORKS_WHILE, {op1, op2, op3}, {op4}); in addWhileOperation()
1316 void createReluStaticWhileModel(std::vector<WrapperModel>* extraModels, WrapperModel* mainModel) { in createReluStaticWhileModel() argument
1320 addReluOperation(mainModel, &modelInputIndexes, &modelOutputIndexes, /*dynamicRank=*/false); in createReluStaticWhileModel()
1322 addWhileOperation(extraModels, mainModel, &modelInputIndexes, &modelOutputIndexes, in createReluStaticWhileModel()
1325 mainModel->identifyInputsAndOutputs(modelInputIndexes, modelOutputIndexes); in createReluStaticWhileModel()
1326 mainModel->finish(); in createReluStaticWhileModel()
1327 ASSERT_TRUE(mainModel->isValid()); in createReluStaticWhileModel()
1373 WrapperModel* mainModel) { in createStaticWhileDynamicWhileModel() argument
1377 addWhileOperation(extraModels, mainModel, &modelInputIndexes, &modelOutputIndexes, in createStaticWhileDynamicWhileModel()
1380 addWhileOperation(extraModels, mainModel, &modelInputIndexes, &modelOutputIndexes, in createStaticWhileDynamicWhileModel()
1383 mainModel->identifyInputsAndOutputs(modelInputIndexes, modelOutputIndexes); in createStaticWhileDynamicWhileModel()
1384 mainModel->finish(); in createStaticWhileDynamicWhileModel()
1385 ASSERT_TRUE(mainModel->isValid()); in createStaticWhileDynamicWhileModel()