Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/runtime/test/
DTestPartitioning.cpp3164 auto elseModel = createBranchOrBodyModel(dimensionedElse); in createIfModel() local
3171 mainModel->addIfOperation(opnd0, *thenModel, *elseModel, {opnd1, opnd2}, {opnd3}); in createIfModel()
3179 models.push_back(std::move(elseModel)); in createIfModel()
3486 auto elseModel = std::make_unique<PartitioningModel>(); in TEST_F() local
3487 const uint32_t elseOpnd0 = elseModel->addFloatOperand(); in TEST_F()
3488 const uint32_t elseOpnd1 = elseModel->addFloatOperand(); in TEST_F()
3489 const uint32_t elseOpnd2 = elseModel->addOperation2To1V1_0(1, elseOpnd0, elseOpnd1); in TEST_F()
3490 elseModel->identifyInputsAndOutputs({elseOpnd0, elseOpnd1}, {elseOpnd2}); in TEST_F()
3491 elseModel->finish(); in TEST_F()
3492 EXPECT_TRUE(elseModel->isValid()); in TEST_F()
[all …]
DTestValidateOperations.cpp4483 const ANeuralNetworksModel* elseModel, bool testMutations) { in testIf() argument
4495 test.setInputOperandValueFromModel(kElseOperand, elseModel); in testIf()
4510 ANeuralNetworksModel* elseModel = makeIdentityModel(&elseDataType); in testIf() local
4511 testIf(outerDims, thenModel, elseModel, testMutations); in testIf()
4513 ANeuralNetworksModel_free(elseModel); in testIf()
/packages/modules/NeuralNetworks/runtime/
DExecutionPlan.cpp2226 const ModelBuilder* elseModel = getReferencedModel(elseOperand); in partitionTheWorkInternal() local
2228 uint32_t elseModelIndex = sourceModels->addModel(elseModel); in partitionTheWorkInternal()
2245 NN_RETURN_IF_ERROR(elseModel->partitionTheWorkInternal( in partitionTheWorkInternal()
2268 for (uint32_t i = 0, n = elseModel->inputCount(); i < n; ++i) { in partitionTheWorkInternal()
2270 elseModelIndex, elseModel->getInputOperandIndex(i)); in partitionTheWorkInternal()
2272 for (uint32_t i = 0, n = elseModel->outputCount(); i < n; ++i) { in partitionTheWorkInternal()
2274 elseModelIndex, elseModel->getOutputOperandIndex(i)); in partitionTheWorkInternal()
2375 const ModelBuilder* elseModel = getReferencedModel(elseOperand); in getPerformance() local
2378 elseModel->getPerformance(preference, device)); in getPerformance()
2432 const ModelBuilder* elseModel = getReferencedModel(elseOperand); in isControlFlowOperationWithOperandOfUnknownSize() local
[all …]