Searched refs:refModel (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/runtime/test/ |
| D | GeneratedTestUtils.cpp | 132 Model* refModel = &refModels[refIndex]; in createModelFromSubgraph() local
133 if (!refModel->isFinished()) { in createModelFromSubgraph()
135 memory, memoryOffset, refModel, refModels); in createModelFromSubgraph()
136 ASSERT_EQ(refModel->finish(), Result::NO_ERROR); in createModelFromSubgraph()
137 ASSERT_TRUE(refModel->isValid()); in createModelFromSubgraph()
139 model->setOperandValueFromModel(index, refModel); in createModelFromSubgraph()
|
| /frameworks/ml/nn/runtime/ |
| D | ModelBuilder.cpp | 877 uint32_t addSubgraph(const ModelBuilder* refModel);
878 void updateOperandLocations(const ModelBuilder* refModel, Subgraph* subgraph);
923 void ModelBuilder::HidlModelMaker::updateOperandLocations(const ModelBuilder* refModel, in updateOperandLocations() argument
933 memcpy(&mOperandValues[offset], &refModel->mSmallOperandValues[originalOffset], in updateOperandLocations()
938 operand.location.poolIndex = mMemories.add(refModel->mMemories[originalPoolIndex]); in updateOperandLocations()
947 operand.location.offset = addSubgraph(refModel->mReferencedModels[refModelIndex]); in updateOperandLocations()
952 uint32_t ModelBuilder::HidlModelMaker::addSubgraph(const ModelBuilder* refModel) { in addSubgraph() argument
954 mRefSubgraphs.push_back(makeSubgraph(refModel)); in addSubgraph()
955 updateOperandLocations(refModel, &mRefSubgraphs.back()); in addSubgraph()
971 for (const auto& refModel : model->mReferencedModels) { in addExtensions() local
[all …]
|