| /frameworks/ml/nn/runtime/test/ |
| D | GeneratedTestUtils.cpp | 90 auto processSubgraph = [&size](const TestSubgraph& subgraph) { in createConstantReferenceMemory() argument
91 for (const TestOperand& operand : subgraph.operands) { in createConstantReferenceMemory()
99 for (const TestSubgraph& subgraph : testModel.referenced) { in createConstantReferenceMemory() local
100 processSubgraph(subgraph); in createConstantReferenceMemory()
105 static void createModelFromSubgraph(const TestSubgraph& subgraph, bool testDynamicOutputShape, in createModelFromSubgraph() argument
110 for (const auto& operand : subgraph.operands) { in createModelFromSubgraph()
150 for (const auto& operation : subgraph.operations) { in createModelFromSubgraph()
155 model->identifyInputsAndOutputs(subgraph.inputIndexes, subgraph.outputIndexes); in createModelFromSubgraph()
|
| D | TestPartitioning.cpp | 282 uint32_t lookupOperation(const HidlModel& model, const Subgraph& subgraph, in lookupOperation() argument
285 [&subgraph](uint32_t index) -> const Operation& { return subgraph.operations[index]; }, in lookupOperation()
286 [&subgraph](uint32_t index) -> const Operand& { return subgraph.operands[index]; }, in lookupOperation()
434 const Subgraph& subgraph) { in getSupportedOperationsForSubgraph() argument
435 CHECK(&subgraph == &model.main || in getSupportedOperationsForSubgraph()
437 [&subgraph](const Subgraph& refSubgraph) { in getSupportedOperationsForSubgraph()
438 return &subgraph == &refSubgraph; in getSupportedOperationsForSubgraph()
440 auto supportsEntireSubgraph = [this, &model, &subgraph](uint32_t refSubgraphOperandIndex) { in getSupportedOperationsForSubgraph()
441 CHECK_LT(refSubgraphOperandIndex, subgraph.operands.size()); in getSupportedOperationsForSubgraph()
442 const Operand& refSubgraphOperand = subgraph.operands[refSubgraphOperandIndex]; in getSupportedOperationsForSubgraph()
[all …]
|
| /frameworks/ml/nn/tools/test_generator/test_harness/include/ |
| D | TestHarness.h | 319 for (const TestSubgraph& subgraph : referenced) { in forEachSubgraph() local
320 handler(subgraph); in forEachSubgraph()
326 for (TestSubgraph& subgraph : referenced) { in forEachSubgraph()
327 handler(subgraph); in forEachSubgraph()
334 newTestModel.forEachSubgraph([](TestSubgraph& subgraph) { in copy()
335 for (TestOperand& operand : subgraph.operands) { in copy()
344 forEachSubgraph([&result](const TestSubgraph& subgraph) { in hasQuant8CoupledOperands()
348 for (const TestOperation& operation : subgraph.operations) { in hasQuant8CoupledOperands()
373 if (subgraph.operands[operandIndex].type == in hasQuant8CoupledOperands()
380 if (subgraph.operands[operandIndex].type == in hasQuant8CoupledOperands()
[all …]
|
| /frameworks/ml/nn/tools/test_generator/ |
| D | example_generator.py | 128 def GetSubgraphStruct(subgraph): argument
131 COMMENT_KEY: subgraph.name,
132 "operands": [GetOperandStruct(op) for op in subgraph.operands],
133 "operations": [GetOperationStruct(op) for op in subgraph.operations],
134 "inputIndexes": [op.model_index for op in subgraph.GetInputs()],
135 "outputIndexes": [op.model_index for op in subgraph.GetOutputs()],
|
| /frameworks/ml/nn/common/ |
| D | ValidateHal.cpp | 471 const Subgraph& subgraph = *getSubgraph(modelOperand); in validateOperations() local
472 CHECK_LT(subgraph.inputIndexes[index], subgraph.operands.size()); in validateOperations()
473 return &subgraph.operands[subgraph.inputIndexes[index]]; in validateOperations()
477 const Subgraph& subgraph = *getSubgraph(modelOperand); in validateOperations() local
478 CHECK_LT(subgraph.outputIndexes[index], subgraph.operands.size()); in validateOperations()
479 return &subgraph.operands[subgraph.outputIndexes[index]]; in validateOperations()
668 static bool checkNoReferenceCycles(const V1_3::Model& model, const V1_3::Subgraph& subgraph, in checkNoReferenceCycles() argument
670 auto [_, isNew] = path->insert(&subgraph); in checkNoReferenceCycles()
675 for (const Operand& operand : subgraph.operands) { in checkNoReferenceCycles()
683 path->erase(&subgraph); in checkNoReferenceCycles()
[all …]
|
| D | CpuExecutor.cpp | 682 int CpuExecutor::executeSubgraph(const Subgraph& subgraph, RunTimeOperandInfo* operands) { in executeSubgraph() argument
683 VLOG(CPUEXE) << "CpuExecutor::executeSubgraph " << toString(subgraph); in executeSubgraph()
685 for (const auto& operation : subgraph.operations) { in executeSubgraph()
691 std::vector<RunTimeOperandInfo> CpuExecutor::initializeRunTimeInfo(const Subgraph& subgraph) { in initializeRunTimeInfo() argument
693 const size_t count = subgraph.operands.size(); in initializeRunTimeInfo()
696 const Operand& from = subgraph.operands[i]; in initializeRunTimeInfo()
|
| D | Utils.cpp | 474 static void logSubgraphToInfo(std::string label, const V1_3::Subgraph& subgraph) { in logSubgraphToInfo() argument
475 LOG(INFO) << label << ".operands" << toString(subgraph.operands); in logSubgraphToInfo()
476 LOG(INFO) << label << ".operations" << toString(subgraph.operations); in logSubgraphToInfo()
477 LOG(INFO) << label << ".inputIndexes" << toString(subgraph.inputIndexes); in logSubgraphToInfo()
478 LOG(INFO) << label << ".outputIndexes" << toString(subgraph.outputIndexes); in logSubgraphToInfo()
|
| /frameworks/ml/nn/runtime/ |
| D | ModelBuilder.cpp | 878 void updateOperandLocations(const ModelBuilder* refModel, Subgraph* subgraph);
915 Subgraph subgraph; in makeSubgraph() local
916 subgraph.operands = model->mOperands; in makeSubgraph()
917 subgraph.operations = model->mOperations; in makeSubgraph()
918 subgraph.inputIndexes = model->mInputIndexes; in makeSubgraph()
919 subgraph.outputIndexes = model->mOutputIndexes; in makeSubgraph()
920 return subgraph; in makeSubgraph()
924 Subgraph* subgraph) { in updateOperandLocations() argument
925 for (Operand& operand : subgraph->operands) { in updateOperandLocations()
942 for (Operand& operand : subgraph->operands) { in updateOperandLocations()
|
| /frameworks/ml/nn/tools/test_generator/test_harness/ |
| D | TestHarness.cpp | 276 model.forEachSubgraph([&hasFloat16Inputs](const TestSubgraph& subgraph) { in checkResults() argument
278 hasFloat16Inputs = std::any_of(subgraph.operands.begin(), subgraph.operands.end(), in checkResults()
296 auto processSubgraph = [](TestSubgraph* subgraph) { in convertQuant8AsymmOperandsToSigned() argument
297 for (TestOperand& operand : subgraph->operands) { in convertQuant8AsymmOperandsToSigned()
312 for (TestSubgraph& subgraph : converted.referenced) { in convertQuant8AsymmOperandsToSigned()
313 processSubgraph(&subgraph); in convertQuant8AsymmOperandsToSigned()
|
| /frameworks/ml/nn/common/include/ |
| D | CpuExecutor.h | 163 std::vector<RunTimeOperandInfo> initializeRunTimeInfo(const hal::Subgraph& subgraph);
171 int executeSubgraph(const hal::Subgraph& subgraph, RunTimeOperandInfo* operands);
|
| /frameworks/ml/nn/tools/api/ |
| D | types.spec | 63 %define model_or_subgraph subgraph
|