Home
last modified time | relevance | path

Searched refs:bodyModel (Results 1 – 6 of 6) sorted by relevance

/packages/modules/NeuralNetworks/runtime/test/
DTestControlFlow.cpp63 Model bodyModel; in TEST_F() local
65 uint32_t i = bodyModel.addOperand(&counterType); in TEST_F()
66 uint32_t n = bodyModel.addOperand(&counterType); in TEST_F()
67 uint32_t one = bodyModel.addConstantOperand(&counterType, 1.0f); in TEST_F()
68 uint32_t noActivation = bodyModel.addConstantOperand(&activationType, kNoActivation); in TEST_F()
69 uint32_t iOut = bodyModel.addOperand(&counterType); in TEST_F()
70 bodyModel.addOperation(ANEURALNETWORKS_ADD, {i, one, noActivation}, {iOut}); in TEST_F()
71 bodyModel.identifyInputsAndOutputs({i, n}, {iOut}); in TEST_F()
72 ASSERT_EQ(bodyModel.finish(), Result::NO_ERROR); in TEST_F()
73 ASSERT_TRUE(bodyModel.isValid()); in TEST_F()
[all …]
DTestFailingDriver.cpp118 WrapperModel bodyModel; in TEST_F() local
120 uint32_t f = bodyModel.addOperand(&floatType); in TEST_F()
121 uint32_t b = bodyModel.addOperand(&boolType); in TEST_F()
122 uint32_t out = bodyModel.addOperand(&floatType); in TEST_F()
123 bodyModel.addOperation(ANEURALNETWORKS_CAST, {f}, {out}); in TEST_F()
124 bodyModel.identifyInputsAndOutputs({f, b}, {out}); in TEST_F()
125 ASSERT_EQ(bodyModel.finish(), Result::NO_ERROR); in TEST_F()
126 ASSERT_TRUE(bodyModel.isValid()); in TEST_F()
136 uint32_t body = model.addModelOperand(&bodyModel); in TEST_F()
DTestPartitioning.cpp835 void addWhileOperation(const PartitioningModel& condModel, const PartitioningModel& bodyModel, in addWhileOperation() argument
839 const uint32_t bodyOperand = addRefModelOperand(bodyModel); in addWhileOperation()
3191 auto bodyModel = createBranchOrBodyModel(dimensionedBody); in createWhileModel() local
3197 mainModel->addWhileOperation(*condModel, *bodyModel, {opnd0, opnd1}, {opnd2}); in createWhileModel()
3205 models.push_back(std::move(bodyModel)); in createWhileModel()
3541 auto bodyModel = std::make_unique<PartitioningModel>(); in TEST_F() local
3542 const uint32_t bodyOpnd0 = bodyModel->addFloatOperand(); in TEST_F()
3543 const uint32_t bodyOpnd1 = bodyModel->addFloatOperand(); in TEST_F()
3544 const uint32_t bodyOpnd2 = bodyModel->addFloatOperand(); in TEST_F()
3545 const uint32_t bodyOpnd3 = bodyModel->addOperation2To1V1_0(1, bodyOpnd0, bodyOpnd1); in TEST_F()
[all …]
DTestIntrospectionControl.cpp1296 WrapperModel* bodyModel = &extraModels->at(extraModels->size() - 1); in addWhileOperation() local
1298 createReluModel(bodyModel, dynamicRank); in addWhileOperation()
1300 ASSERT_TRUE(bodyModel->isValid()); in addWhileOperation()
1308 mainModel->setOperandValueFromModel(op2, bodyModel); in addWhileOperation()
DTestValidateOperations.cpp4591 const ANeuralNetworksModel* bodyModel, bool testMutations) { in testWhile() argument
4600 test.setInputOperandValueFromModel(kBodyOperand, bodyModel); in testWhile()
4618 ANeuralNetworksModel* bodyModel = makeWhileBodyModel(&bodyDataType); in testWhile() local
4619 testWhile(outerDims, condModel, bodyModel, testMutations); in testWhile()
4621 ANeuralNetworksModel_free(bodyModel); in testWhile()
/packages/modules/NeuralNetworks/runtime/
DExecutionPlan.cpp2283 const ModelBuilder* bodyModel = getReferencedModel(bodyOperand); in partitionTheWorkInternal() local
2285 uint32_t bodyModelIndex = sourceModels->addModel(bodyModel); in partitionTheWorkInternal()
2304 NN_RETURN_IF_ERROR(bodyModel->partitionTheWorkInternal( in partitionTheWorkInternal()
2327 for (uint32_t i = 0, n = bodyModel->inputCount(); i < n; ++i) { in partitionTheWorkInternal()
2329 bodyModelIndex, bodyModel->getInputOperandIndex(i)); in partitionTheWorkInternal()
2331 for (uint32_t i = 0, n = bodyModel->outputCount(); i < n; ++i) { in partitionTheWorkInternal()
2333 bodyModelIndex, bodyModel->getOutputOperandIndex(i)); in partitionTheWorkInternal()
2386 const ModelBuilder* bodyModel = getReferencedModel(bodyOperand); in getPerformance() local
2389 bodyModel->getPerformance(preference, device); in getPerformance()
2446 const ModelBuilder* bodyModel = getReferencedModel(bodyOperand); in isControlFlowOperationWithOperandOfUnknownSize() local
[all …]