Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/runtime/test/
H A DTestPartitioning.cpp3178 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()
3483 thenModel->finish(); in TEST_F()
3484 EXPECT_TRUE(thenModel->isValid()); in TEST_F()
3614 auto thenModel = std::make_unique<PartitioningModel>(); in TEST_F() local
3615 const uint32_t thenOpnd0 = thenModel->addFloatOperand(); in TEST_F()
3616 const uint32_t thenOpnd1 = thenModel->addFloatOperand(); in TEST_F()
3619 thenModel->finish(); in TEST_F()
[all …]
H A 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()
/aosp12/packages/modules/NeuralNetworks/runtime/
H A 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()
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
2435 containsUnknownSize(thenModel, thenModel->getInputOperandIndexes()) || in isControlFlowOperationWithOperandOfUnknownSize()
[all …]