Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/runtime/test/
DTestPartitioning.cpp3163 auto thenModel = createBranchOrBodyModel(dimensionedThen); in createIfModel() local
3171 mainModel->addIfOperation(opnd0, *thenModel, *elseModel, {opnd1, opnd2}, {opnd3}); in createIfModel()
3178 models.push_back(std::move(thenModel)); in createIfModel()
3478 auto thenModel = std::make_unique<PartitioningModel>(); in TEST_F() local
3479 const uint32_t thenOpnd0 = thenModel->addFloatOperand(); in TEST_F()
3480 const uint32_t thenOpnd1 = thenModel->addFloatOperand(); in TEST_F()
3481 const uint32_t thenOpnd2 = thenModel->addOperation2To1V1_0(0, thenOpnd0, thenOpnd1); in TEST_F()
3482 thenModel->identifyInputsAndOutputs({thenOpnd0, thenOpnd1}, {thenOpnd2}); in TEST_F()
3483 thenModel->finish(); in TEST_F()
3484 EXPECT_TRUE(thenModel->isValid()); in TEST_F()
[all …]
DTestValidateOperations.cpp4482 void testIf(const std::vector<uint32_t>& outerDims, const ANeuralNetworksModel* thenModel, in testIf() argument
4494 test.setInputOperandValueFromModel(kThenOperand, thenModel); in testIf()
4509 ANeuralNetworksModel* thenModel = makeIdentityModel(&thenDataType); in testIf() local
4511 testIf(outerDims, thenModel, elseModel, testMutations); in testIf()
4512 ANeuralNetworksModel_free(thenModel); in testIf()
/packages/modules/NeuralNetworks/runtime/
DExecutionPlan.cpp2225 const ModelBuilder* thenModel = getReferencedModel(thenOperand); in partitionTheWorkInternal() local
2227 uint32_t thenModelIndex = sourceModels->addModel(thenModel); in partitionTheWorkInternal()
2241 NN_RETURN_IF_ERROR(thenModel->partitionTheWorkInternal( in partitionTheWorkInternal()
2259 for (uint32_t i = 0, n = thenModel->inputCount(); i < n; ++i) { in partitionTheWorkInternal()
2261 thenModelIndex, thenModel->getInputOperandIndex(i)); in partitionTheWorkInternal()
2263 for (uint32_t i = 0, n = thenModel->outputCount(); i < n; ++i) { in partitionTheWorkInternal()
2265 thenModelIndex, thenModel->getOutputOperandIndex(i)); in partitionTheWorkInternal()
2374 const ModelBuilder* thenModel = getReferencedModel(thenOperand); in getPerformance() local
2377 0.5 * (thenModel->getPerformance(preference, device) + in getPerformance()
2431 const ModelBuilder* thenModel = getReferencedModel(thenOperand); in isControlFlowOperationWithOperandOfUnknownSize() local
[all …]