Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/runtime/test/
DTestIntrospectionControl.cpp1287 void addWhileOperation(std::vector<WrapperModel>* extraModels, WrapperModel* mainModel, in addWhileOperation() argument
1304 const uint32_t op1 = mainModel->addOperand(&modelType); in addWhileOperation()
1305 const uint32_t op2 = mainModel->addOperand(&modelType); in addWhileOperation()
1306 const uint32_t op3 = mainModel->addOperand(&floatType); in addWhileOperation()
1307 const uint32_t op4 = mainModel->addOperand(&floatType); in addWhileOperation()
1308 mainModel->setOperandValueFromModel(op1, condModel); in addWhileOperation()
1309 mainModel->setOperandValueFromModel(op2, bodyModel); in addWhileOperation()
1311 mainModel->addOperation(ANEURALNETWORKS_WHILE, {op1, op2, op3}, {op4}); in addWhileOperation()
1317 void createReluStaticWhileModel(std::vector<WrapperModel>* extraModels, WrapperModel* mainModel) { in createReluStaticWhileModel() argument
1321 addReluOperation(mainModel, &modelInputIndexes, &modelOutputIndexes, /*dynamicRank=*/false); in createReluStaticWhileModel()
[all …]
DTestPartitioning.cpp2751 auto mainModel = std::make_unique<PartitioningModel>(); in createControlFlowModelForCachingTests() local
2753 const uint32_t opnd0 = mainModel->addBooleanOperand(); in createControlFlowModelForCachingTests()
2754 const uint32_t opnd1 = mainModel->addFloatOperand(); in createControlFlowModelForCachingTests()
2755 const uint32_t opnd2 = mainModel->addFloatOperand(); in createControlFlowModelForCachingTests()
2756 const uint32_t opnd3 = mainModel->addFloatOperand(); in createControlFlowModelForCachingTests()
2757 mainModel->addIfOperation(opnd0, *trueModel, *falseModel, {opnd1, opnd2}, {opnd3}); in createControlFlowModelForCachingTests()
2758 mainModel->identifyInputsAndOutputs({opnd0, opnd1, opnd2}, {opnd3}); in createControlFlowModelForCachingTests()
2759 mainModel->finish(); in createControlFlowModelForCachingTests()
2760 ASSERT_TRUE(mainModel->isValid()); in createControlFlowModelForCachingTests()
2764 models->push_back(std::move(mainModel)); in createControlFlowModelForCachingTests()
[all …]
/packages/modules/NeuralNetworks/runtime/
DExecutionPlan.cpp784 int ExecutionStep::finishStepModel(const ModelBuilder* mainModel, bool* hasOutputOfUnknownSize, in finishStepModel() argument
797 mStepModel.relaxComputationFloat32toFloat16(mainModel->isComputationFloat32RelaxedToFloat16()); in finishStepModel()
820 for (uint32_t i = 0, n = mainModel->inputCount(); i < n; ++i) { in finishStepModel()
821 mainModelOperandToInputIndex[mainModel->getInputOperandIndex(i)] = i; in finishStepModel()
824 for (uint32_t i = 0, n = mainModel->outputCount(); i < n; ++i) { in finishStepModel()
825 mainModelOperandToOutputIndex[mainModel->getOutputOperandIndex(i)] = i; in finishStepModel()
923 const ModelBuilder* mainModel = sourceModels->getModel(kMainModelInSourceModels); in finish() local
940 int n = step->finishStepModel(mainModel, &stepHasDynamicTemporaries, in finish()
993 for (uint32_t i = 0, n = mainModel->inputCount(); i < n; ++i) { in finish()
994 SourceOperandIndex index(kMainModelInSourceModels, mainModel->getInputOperandIndex(i)); in finish()
[all …]
DModelBuilder.cpp904 Model makeModel(const ModelBuilder* mainModel);
932 Model ModelBuilder::ModelMaker::makeModel(const ModelBuilder* mainModel) { in makeModel() argument
933 addExtensions(mainModel); in makeModel()
935 model.main = makeSubgraph(mainModel); in makeModel()
936 updateOperandLocations(mainModel, &model.main); in makeModel()
942 model.relaxComputationFloat32toFloat16 = mainModel->mRelaxComputationFloat32toFloat16; in makeModel()
DExecutionPlan.h277 int finishStepModel(const ModelBuilder* mainModel, bool* hasOutputOfUnknownSize,