Home
last modified time | relevance | path

Searched refs:condModel (Results 1 – 4 of 4) sorted by relevance

/packages/modules/NeuralNetworks/runtime/test/
DTestPartitioning.cpp835 void addWhileOperation(const PartitioningModel& condModel, const PartitioningModel& bodyModel, in addWhileOperation() argument
838 const uint32_t condOperand = addRefModelOperand(condModel); in addWhileOperation()
3190 auto condModel = createCondModel(dimensionedCond); in createWhileModel() local
3197 mainModel->addWhileOperation(*condModel, *bodyModel, {opnd0, opnd1}, {opnd2}); in createWhileModel()
3204 models.push_back(std::move(condModel)); in createWhileModel()
3527 auto condModel = std::make_unique<PartitioningModel>(); in TEST_F() local
3528 const uint32_t condOpnd0 = condModel->addFloatOperand(); in TEST_F()
3529 const uint32_t condOpnd1 = condModel->addFloatOperand(); in TEST_F()
3530 const uint32_t condOpnd2 = condModel->addFloatOperand(); in TEST_F()
3531 const uint32_t condOpnd3 = condModel->addOperation2To1V1_0(0, condOpnd0, condOpnd1); in TEST_F()
[all …]
DTestIntrospectionControl.cpp1295 WrapperModel* condModel = &extraModels->at(extraModels->size() - 2); in addWhileOperation() local
1297 createCondModel(condModel, dynamicRank); in addWhileOperation()
1299 ASSERT_TRUE(condModel->isValid()); in addWhileOperation()
1307 mainModel->setOperandValueFromModel(op1, condModel); in addWhileOperation()
DTestValidateOperations.cpp4590 void testWhile(const std::vector<uint32_t>& outerDims, const ANeuralNetworksModel* condModel, in testWhile() argument
4599 test.setInputOperandValueFromModel(kCondOperand, condModel); in testWhile()
4617 ANeuralNetworksModel* condModel = makeWhileCondModel(&condDataType, &boolType); in testWhile() local
4619 testWhile(outerDims, condModel, bodyModel, testMutations); in testWhile()
4620 ANeuralNetworksModel_free(condModel); in testWhile()
/packages/modules/NeuralNetworks/runtime/
DExecutionPlan.cpp2282 const ModelBuilder* condModel = getReferencedModel(condOperand); in partitionTheWorkInternal() local
2284 uint32_t condModelIndex = sourceModels->addModel(condModel); in partitionTheWorkInternal()
2299 NN_RETURN_IF_ERROR(condModel->partitionTheWorkInternal( in partitionTheWorkInternal()
2320 for (uint32_t i = 0, n = condModel->inputCount(); i < n; ++i) { in partitionTheWorkInternal()
2322 condModelIndex, condModel->getInputOperandIndex(i)); in partitionTheWorkInternal()
2325 SourceOperandIndex(condModelIndex, condModel->getOutputOperandIndex(0)); in partitionTheWorkInternal()
2385 const ModelBuilder* condModel = getReferencedModel(condOperand); in getPerformance() local
2388 condModel->getPerformance(preference, device) + in getPerformance()
2445 const ModelBuilder* condModel = getReferencedModel(condOperand); in isControlFlowOperationWithOperandOfUnknownSize() local
2449 containsUnknownSize(condModel, condModel->getInputOperandIndexes()) || in isControlFlowOperationWithOperandOfUnknownSize()
[all …]