Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/runtime/test/
DTestIntrospectionControl.cpp1286 void addWhileOperation(std::vector<WrapperModel>* extraModels, WrapperModel* mainModel, in addWhileOperation() argument
1303 const uint32_t op1 = mainModel->addOperand(&modelType); in addWhileOperation()
1304 const uint32_t op2 = mainModel->addOperand(&modelType); in addWhileOperation()
1305 const uint32_t op3 = mainModel->addOperand(&floatType); in addWhileOperation()
1306 const uint32_t op4 = mainModel->addOperand(&floatType); in addWhileOperation()
1307 mainModel->setOperandValueFromModel(op1, condModel); in addWhileOperation()
1308 mainModel->setOperandValueFromModel(op2, bodyModel); in addWhileOperation()
1310 mainModel->addOperation(ANEURALNETWORKS_WHILE, {op1, op2, op3}, {op4}); in addWhileOperation()
1316 void createReluStaticWhileModel(std::vector<WrapperModel>* extraModels, WrapperModel* mainModel) { in createReluStaticWhileModel() argument
1320 addReluOperation(mainModel, &modelInputIndexes, &modelOutputIndexes, /*dynamicRank=*/false); in createReluStaticWhileModel()
[all …]
DTestPartitioning.cpp2750 auto mainModel = std::make_unique<PartitioningModel>(); in createControlFlowModelForCachingTests() local
2752 const uint32_t opnd0 = mainModel->addBooleanOperand(); in createControlFlowModelForCachingTests()
2753 const uint32_t opnd1 = mainModel->addFloatOperand(); in createControlFlowModelForCachingTests()
2754 const uint32_t opnd2 = mainModel->addFloatOperand(); in createControlFlowModelForCachingTests()
2755 const uint32_t opnd3 = mainModel->addFloatOperand(); in createControlFlowModelForCachingTests()
2756 mainModel->addIfOperation(opnd0, *trueModel, *falseModel, {opnd1, opnd2}, {opnd3}); in createControlFlowModelForCachingTests()
2757 mainModel->identifyInputsAndOutputs({opnd0, opnd1, opnd2}, {opnd3}); in createControlFlowModelForCachingTests()
2758 mainModel->finish(); in createControlFlowModelForCachingTests()
2759 ASSERT_TRUE(mainModel->isValid()); in createControlFlowModelForCachingTests()
2763 models->push_back(std::move(mainModel)); in createControlFlowModelForCachingTests()
[all …]
/packages/modules/NeuralNetworks/runtime/
DExecutionPlan.cpp744 int ExecutionStep::finishStepModel(const ModelBuilder* mainModel, bool* hasOutputOfUnknownSize, in finishStepModel() argument
757 mStepModel.relaxComputationFloat32toFloat16(mainModel->isComputationFloat32RelaxedToFloat16()); in finishStepModel()
780 for (uint32_t i = 0, n = mainModel->inputCount(); i < n; ++i) { in finishStepModel()
781 mainModelOperandToInputIndex[mainModel->getInputOperandIndex(i)] = i; in finishStepModel()
784 for (uint32_t i = 0, n = mainModel->outputCount(); i < n; ++i) { in finishStepModel()
785 mainModelOperandToOutputIndex[mainModel->getOutputOperandIndex(i)] = i; in finishStepModel()
880 const ModelBuilder* mainModel = sourceModels->getModel(kMainModelInSourceModels); in finish() local
897 int n = step->finishStepModel(mainModel, &stepHasDynamicTemporaries, in finish()
949 for (uint32_t i = 0, n = mainModel->inputCount(); i < n; ++i) { in finish()
950 SourceOperandIndex index(kMainModelInSourceModels, mainModel->getInputOperandIndex(i)); in finish()
[all …]
DModelBuilder.cpp874 Model makeModel(const ModelBuilder* mainModel);
897 Model ModelBuilder::ModelMaker::makeModel(const ModelBuilder* mainModel) { in makeModel() argument
898 addExtensions(mainModel); in makeModel()
900 model.main = makeSubgraph(mainModel); in makeModel()
901 updateOperandLocations(mainModel, &model.main); in makeModel()
907 model.relaxComputationFloat32toFloat16 = mainModel->mRelaxComputationFloat32toFloat16; in makeModel()
DExecutionPlan.h277 int finishStepModel(const ModelBuilder* mainModel, bool* hasOutputOfUnknownSize,