Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/runtime/test/
DTestPartitioning.cpp3165 auto thenModel = createBranchOrBodyModel(dimensionedThen); in createIfModel() local
3173 mainModel->addIfOperation(opnd0, *thenModel, *elseModel, {opnd1, opnd2}, {opnd3}); in createIfModel()
3180 models.push_back(std::move(thenModel)); in createIfModel()
3480 auto thenModel = std::make_unique<PartitioningModel>(); in TEST_F() local
3481 const uint32_t thenOpnd0 = thenModel->addFloatOperand(); in TEST_F()
3482 const uint32_t thenOpnd1 = thenModel->addFloatOperand(); in TEST_F()
3483 const uint32_t thenOpnd2 = thenModel->addOperation2To1V1_0(0, thenOpnd0, thenOpnd1); in TEST_F()
3484 thenModel->identifyInputsAndOutputs({thenOpnd0, thenOpnd1}, {thenOpnd2}); in TEST_F()
3485 thenModel->finish(); in TEST_F()
3486 EXPECT_TRUE(thenModel->isValid()); in TEST_F()
[all …]
DTestValidateOperations.cpp4551 void testIf(const std::vector<uint32_t>& outerDims, const ANeuralNetworksModel* thenModel, in testIf() argument
4563 test.setInputOperandValueFromModel(kThenOperand, thenModel); in testIf()
4578 ANeuralNetworksModel* thenModel = makeIdentityModel(&thenDataType); in testIf() local
4580 testIf(outerDims, thenModel, elseModel, testMutations); in testIf()
4581 ANeuralNetworksModel_free(thenModel); in testIf()
/packages/modules/NeuralNetworks/runtime/
DExecutionPlan.cpp2278 const ModelBuilder* thenModel = getReferencedModel(thenOperand); in partitionTheWorkInternal() local
2280 uint32_t thenModelIndex = sourceModels->addModel(thenModel); in partitionTheWorkInternal()
2294 NN_RETURN_IF_ERROR(thenModel->partitionTheWorkInternal( in partitionTheWorkInternal()
2312 for (uint32_t i = 0, n = thenModel->inputCount(); i < n; ++i) { in partitionTheWorkInternal()
2314 thenModelIndex, thenModel->getInputOperandIndex(i)); in partitionTheWorkInternal()
2316 for (uint32_t i = 0, n = thenModel->outputCount(); i < n; ++i) { in partitionTheWorkInternal()
2318 thenModelIndex, thenModel->getOutputOperandIndex(i)); in partitionTheWorkInternal()
2427 const ModelBuilder* thenModel = getReferencedModel(thenOperand); in getPerformance() local
2430 0.5 * (thenModel->getPerformance(preference, device) + in getPerformance()
2484 const ModelBuilder* thenModel = getReferencedModel(thenOperand); in isControlFlowOperationWithOperandOfUnknownSize() local
[all …]