Searched refs:condModel (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/runtime/ |
| D | ExecutionPlan.cpp | 1787 const ModelBuilder* condModel = getReferencedModel(condOperand); in partitionTheWorkInternal() local
1789 uint32_t condModelIndex = sourceModels->addModel(condModel); in partitionTheWorkInternal()
1804 NN_RETURN_IF_ERROR(condModel->partitionTheWorkInternal( in partitionTheWorkInternal()
1825 for (uint32_t i = 0, n = condModel->inputCount(); i < n; ++i) { in partitionTheWorkInternal()
1827 condModelIndex, condModel->getInputOperandIndex(i)); in partitionTheWorkInternal()
1830 SourceOperandIndex(condModelIndex, condModel->getOutputOperandIndex(0)); in partitionTheWorkInternal()
1890 const ModelBuilder* condModel = getReferencedModel(condOperand); in getPerformance() local
1893 condModel->getPerformance(preference, device) + in getPerformance()
1950 const ModelBuilder* condModel = getReferencedModel(condOperand); in isControlFlowOperationWithOperandOfUnknownSize() local
1954 containsUnknownSize(condModel, condModel->getInputOperandIndexes()) || in isControlFlowOperationWithOperandOfUnknownSize()
[all …]
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestPartitioning.cpp | 785 void addWhileOperation(const PartitioningModel& condModel, const PartitioningModel& bodyModel, in addWhileOperation() argument
788 const uint32_t condOperand = addRefModelOperand(condModel); in addWhileOperation()
2628 auto condModel = createCondModel(dimensionedCond); in createWhileModel() local
2635 mainModel->addWhileOperation(*condModel, *bodyModel, {opnd0, opnd1}, {opnd2}); in createWhileModel()
2642 models.push_back(std::move(condModel)); in createWhileModel()
|
| D | TestValidateOperations.cpp | 4591 void testWhile(const std::vector<uint32_t>& outerDims, const ANeuralNetworksModel* condModel, in testWhile() argument
4600 test.setInputOperandValueFromModel(kCondOperand, condModel); in testWhile()
4618 ANeuralNetworksModel* condModel = makeWhileCondModel(&condDataType, &boolType); in testWhile() local
4620 testWhile(outerDims, condModel, bodyModel, testMutations); in testWhile()
4621 ANeuralNetworksModel_free(condModel); in testWhile()
|