Home
last modified time | relevance | path

Searched refs:mainModel (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/runtime/test/
DTestPartitioning.cpp2179 auto mainModel = std::make_unique<PartitioningModel>(); in createControlFlowModelForCachingTests() local
2181 const uint32_t opnd0 = mainModel->addBooleanOperand(); in createControlFlowModelForCachingTests()
2182 const uint32_t opnd1 = mainModel->addFloatOperand(); in createControlFlowModelForCachingTests()
2183 const uint32_t opnd2 = mainModel->addFloatOperand(); in createControlFlowModelForCachingTests()
2184 const uint32_t opnd3 = mainModel->addFloatOperand(); in createControlFlowModelForCachingTests()
2185 mainModel->addIfOperation(opnd0, *trueModel, *falseModel, {opnd1, opnd2}, {opnd3}); in createControlFlowModelForCachingTests()
2186 mainModel->identifyInputsAndOutputs({opnd0, opnd1, opnd2}, {opnd3}); in createControlFlowModelForCachingTests()
2187 mainModel->finish(); in createControlFlowModelForCachingTests()
2188 ASSERT_TRUE(mainModel->isValid()); in createControlFlowModelForCachingTests()
2192 models->push_back(std::move(mainModel)); in createControlFlowModelForCachingTests()
[all …]
/frameworks/ml/nn/runtime/
DModelBuilder.cpp876 Model makeHidlModel(const ModelBuilder* mainModel);
899 Model ModelBuilder::HidlModelMaker::makeHidlModel(const ModelBuilder* mainModel) { in makeHidlModel() argument
900 addExtensions(mainModel); in makeHidlModel()
902 model.main = makeSubgraph(mainModel); in makeHidlModel()
903 updateOperandLocations(mainModel, &model.main); in makeHidlModel()
909 model.relaxComputationFloat32toFloat16 = mainModel->mRelaxComputationFloat32toFloat16; in makeHidlModel()
DExecutionPlan.h147 int finishStepModel(const ModelBuilder* mainModel, bool* hasOutputOfUnknownSize,