Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/
DExecutionPlan.cpp1730 const ModelBuilder* thenModel = getReferencedModel(thenOperand); in partitionTheWorkInternal() local
1732 uint32_t thenModelIndex = sourceModels->addModel(thenModel); in partitionTheWorkInternal()
1746 NN_RETURN_IF_ERROR(thenModel->partitionTheWorkInternal( in partitionTheWorkInternal()
1764 for (uint32_t i = 0, n = thenModel->inputCount(); i < n; ++i) { in partitionTheWorkInternal()
1766 thenModelIndex, thenModel->getInputOperandIndex(i)); in partitionTheWorkInternal()
1768 for (uint32_t i = 0, n = thenModel->outputCount(); i < n; ++i) { in partitionTheWorkInternal()
1770 thenModelIndex, thenModel->getOutputOperandIndex(i)); in partitionTheWorkInternal()
1879 const ModelBuilder* thenModel = getReferencedModel(thenOperand); in getPerformance() local
1882 0.5 * (thenModel->getPerformance(preference, device) + in getPerformance()
1936 const ModelBuilder* thenModel = getReferencedModel(thenOperand); in isControlFlowOperationWithOperandOfUnknownSize() local
[all …]
/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp4483 void testIf(const std::vector<uint32_t>& outerDims, const ANeuralNetworksModel* thenModel, in testIf() argument
4495 test.setInputOperandValueFromModel(kThenOperand, thenModel); in testIf()
4510 ANeuralNetworksModel* thenModel = makeIdentityModel(&thenDataType); in testIf() local
4512 testIf(outerDims, thenModel, elseModel, testMutations); in testIf()
4513 ANeuralNetworksModel_free(thenModel); in testIf()
DTestPartitioning.cpp2601 auto thenModel = createBranchOrBodyModel(dimensionedThen); in createIfModel() local
2609 mainModel->addIfOperation(opnd0, *thenModel, *elseModel, {opnd1, opnd2}, {opnd3}); in createIfModel()
2616 models.push_back(std::move(thenModel)); in createIfModel()