Searched refs:condModel (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/runtime/test/ |
| D | TestPartitioning.cpp | 775 void addWhileOperation(const PartitioningModel& condModel, const PartitioningModel& bodyModel, in addWhileOperation() argument
778 const uint32_t condOperand = addRefModelOperand(condModel); in addWhileOperation()
2617 auto condModel = createCondModel(); in createWhileModel() local
2626 mainModel->addWhileOperation(*condModel, *bodyModel, {opnd0, opnd1}, {opnd2}); in createWhileModel()
2633 models.push_back(std::move(condModel)); in createWhileModel()
|
| D | TestValidateOperations.cpp | 4590 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()
|