Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/test/
DTestPartitioning.cpp2188 auto mainModel = std::make_unique<PartitioningModel>(); in createControlFlowModelForCachingTests() local
2190 const uint32_t opnd0 = mainModel->addBooleanOperand(); in createControlFlowModelForCachingTests()
2191 const uint32_t opnd1 = mainModel->addFloatOperand(); in createControlFlowModelForCachingTests()
2192 const uint32_t opnd2 = mainModel->addFloatOperand(); in createControlFlowModelForCachingTests()
2193 const uint32_t opnd3 = mainModel->addFloatOperand(); in createControlFlowModelForCachingTests()
2194 mainModel->addIfOperation(opnd0, *trueModel, *falseModel, {opnd1, opnd2}, {opnd3}); in createControlFlowModelForCachingTests()
2195 mainModel->identifyInputsAndOutputs({opnd0, opnd1, opnd2}, {opnd3}); in createControlFlowModelForCachingTests()
2196 mainModel->finish(); in createControlFlowModelForCachingTests()
2197 ASSERT_TRUE(mainModel->isValid()); in createControlFlowModelForCachingTests()
2201 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.cpp479 int ExecutionStep::finishStepModel(const ModelBuilder* mainModel, bool* hasOutputOfUnknownSize, in finishStepModel() argument
492 mStepModel.relaxComputationFloat32toFloat16(mainModel->isComputationFloat32RelaxedToFloat16()); in finishStepModel()
506 for (uint32_t i = 0, n = mainModel->inputCount(); i < n; ++i) { in finishStepModel()
507 mainModelOperandToInputIndex[mainModel->getInputOperandIndex(i)] = i; in finishStepModel()
510 for (uint32_t i = 0, n = mainModel->outputCount(); i < n; ++i) { in finishStepModel()
511 mainModelOperandToOutputIndex[mainModel->getOutputOperandIndex(i)] = i; in finishStepModel()
616 const ModelBuilder* mainModel = sourceModels->getModel(kMainModelInSourceModels); in finish() local
632 int n = step->finishStepModel(mainModel, &mHasStepModelOutputOfUnknownSize, in finish()
666 for (uint32_t i = 0, n = mainModel->inputCount(); i < n; ++i) { in finish()
667 SourceOperandIndex index(kMainModelInSourceModels, mainModel->getInputOperandIndex(i)); in finish()
[all …]
DExecutionPlan.h147 int finishStepModel(const ModelBuilder* mainModel, bool* hasOutputOfUnknownSize,