Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/
DExecutionPlan.cpp1731 const ModelBuilder* elseModel = getReferencedModel(elseOperand); in partitionTheWorkInternal() local
1733 uint32_t elseModelIndex = sourceModels->addModel(elseModel); in partitionTheWorkInternal()
1750 NN_RETURN_IF_ERROR(elseModel->partitionTheWorkInternal( in partitionTheWorkInternal()
1773 for (uint32_t i = 0, n = elseModel->inputCount(); i < n; ++i) { in partitionTheWorkInternal()
1775 elseModelIndex, elseModel->getInputOperandIndex(i)); in partitionTheWorkInternal()
1777 for (uint32_t i = 0, n = elseModel->outputCount(); i < n; ++i) { in partitionTheWorkInternal()
1779 elseModelIndex, elseModel->getOutputOperandIndex(i)); in partitionTheWorkInternal()
1880 const ModelBuilder* elseModel = getReferencedModel(elseOperand); in getPerformance() local
1883 elseModel->getPerformance(preference, device)); in getPerformance()
1937 const ModelBuilder* elseModel = getReferencedModel(elseOperand); in isControlFlowOperationWithOperandOfUnknownSize() local
[all …]
/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp4484 const ANeuralNetworksModel* elseModel, bool testMutations) { in testIf() argument
4496 test.setInputOperandValueFromModel(kElseOperand, elseModel); in testIf()
4511 ANeuralNetworksModel* elseModel = makeIdentityModel(&elseDataType); in testIf() local
4512 testIf(outerDims, thenModel, elseModel, testMutations); in testIf()
4514 ANeuralNetworksModel_free(elseModel); in testIf()
DTestPartitioning.cpp2602 auto elseModel = createBranchOrBodyModel(dimensionedElse); in createIfModel() local
2609 mainModel->addIfOperation(opnd0, *thenModel, *elseModel, {opnd1, opnd2}, {opnd3}); in createIfModel()
2617 models.push_back(std::move(elseModel)); in createIfModel()