Home
last modified time | relevance | path

Searched refs:bodyModel (Results 1 – 4 of 4) 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.cpp775 void addWhileOperation(const PartitioningModel& condModel, const PartitioningModel& bodyModel, in addWhileOperation() argument
779 const uint32_t bodyOperand = addRefModelOperand(bodyModel); in addWhileOperation()
2618 auto bodyModel = createBranchOrBodyModel(); in createWhileModel() local
2626 mainModel->addWhileOperation(*condModel, *bodyModel, {opnd0, opnd1}, {opnd2}); in createWhileModel()
2634 models.push_back(std::move(bodyModel)); in createWhileModel()
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()