Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/test/
DTestControlFlow.cpp59 Model bodyModel; in TEST_F() local
61 uint32_t i = bodyModel.addOperand(&counterType); in TEST_F()
62 uint32_t n = bodyModel.addOperand(&counterType); in TEST_F()
63 uint32_t one = bodyModel.addConstantOperand(&counterType, 1.0f); in TEST_F()
64 uint32_t noActivation = bodyModel.addConstantOperand(&activationType, kNoActivation); in TEST_F()
65 uint32_t iOut = bodyModel.addOperand(&counterType); in TEST_F()
66 bodyModel.addOperation(ANEURALNETWORKS_ADD, {i, one, noActivation}, {iOut}); in TEST_F()
67 bodyModel.identifyInputsAndOutputs({i, n}, {iOut}); in TEST_F()
68 ASSERT_EQ(bodyModel.finish(), Result::NO_ERROR); in TEST_F()
69 ASSERT_TRUE(bodyModel.isValid()); in TEST_F()
[all …]
DTestFailingDriver.cpp119 WrapperModel bodyModel; in TEST_F() local
121 uint32_t f = bodyModel.addOperand(&floatType); in TEST_F()
122 uint32_t b = bodyModel.addOperand(&boolType); in TEST_F()
123 uint32_t out = bodyModel.addOperand(&floatType); in TEST_F()
124 bodyModel.addOperation(ANEURALNETWORKS_CAST, {f}, {out}); in TEST_F()
125 bodyModel.identifyInputsAndOutputs({f, b}, {out}); in TEST_F()
126 ASSERT_EQ(bodyModel.finish(), Result::NO_ERROR); in TEST_F()
127 ASSERT_TRUE(bodyModel.isValid()); in TEST_F()
137 uint32_t body = model.addModelOperand(&bodyModel); in TEST_F()
DTestPartitioning.cpp785 void addWhileOperation(const PartitioningModel& condModel, const PartitioningModel& bodyModel, in addWhileOperation() argument
789 const uint32_t bodyOperand = addRefModelOperand(bodyModel); in addWhileOperation()
2629 auto bodyModel = createBranchOrBodyModel(dimensionedBody); in createWhileModel() local
2635 mainModel->addWhileOperation(*condModel, *bodyModel, {opnd0, opnd1}, {opnd2}); in createWhileModel()
2643 models.push_back(std::move(bodyModel)); in createWhileModel()
DTestValidateOperations.cpp4592 const ANeuralNetworksModel* bodyModel, bool testMutations) { in testWhile() argument
4601 test.setInputOperandValueFromModel(kBodyOperand, bodyModel); in testWhile()
4619 ANeuralNetworksModel* bodyModel = makeWhileBodyModel(&bodyDataType); in testWhile() local
4620 testWhile(outerDims, condModel, bodyModel, testMutations); in testWhile()
4622 ANeuralNetworksModel_free(bodyModel); in testWhile()
/frameworks/ml/nn/runtime/
DExecutionPlan.cpp1788 const ModelBuilder* bodyModel = getReferencedModel(bodyOperand); in partitionTheWorkInternal() local
1790 uint32_t bodyModelIndex = sourceModels->addModel(bodyModel); in partitionTheWorkInternal()
1809 NN_RETURN_IF_ERROR(bodyModel->partitionTheWorkInternal( in partitionTheWorkInternal()
1832 for (uint32_t i = 0, n = bodyModel->inputCount(); i < n; ++i) { in partitionTheWorkInternal()
1834 bodyModelIndex, bodyModel->getInputOperandIndex(i)); in partitionTheWorkInternal()
1836 for (uint32_t i = 0, n = bodyModel->outputCount(); i < n; ++i) { in partitionTheWorkInternal()
1838 bodyModelIndex, bodyModel->getOutputOperandIndex(i)); in partitionTheWorkInternal()
1891 const ModelBuilder* bodyModel = getReferencedModel(bodyOperand); in getPerformance() local
1894 bodyModel->getPerformance(preference, device); in getPerformance()
1951 const ModelBuilder* bodyModel = getReferencedModel(bodyOperand); in isControlFlowOperationWithOperandOfUnknownSize() local
[all …]