Home
last modified time | relevance | path

Searched defs:subgraph (Results 1 – 21 of 21) sorted by relevance

/packages/modules/NeuralNetworks/driver/sample_hidl/
DSampleDriverFloatXNNPACK.cpp109 const V1_3::Subgraph& subgraph, const std::vector<RunTimePoolInfo>& modelPoolInfos, in initializeRunTimeInfo()
175 std::unique_ptr<xnn_subgraph, decltype(&xnn_delete_subgraph)> subgraph( in Create() local
496 static V1_3::ErrorStatus VisitNode(xnn_subgraph_t subgraph, const V1_3::Operation& operation, in VisitNode()
560 static V1_3::ErrorStatus VisitAbsNode(xnn_subgraph_t subgraph, const V1_3::Operation& operation, in VisitAbsNode()
580 static V1_3::ErrorStatus VisitAddNode(xnn_subgraph_t subgraph, const V1_3::Operation& operation, in VisitAddNode()
610 static V1_3::ErrorStatus VisitAveragePool2DNode(xnn_subgraph_t subgraph, in VisitAveragePool2DNode()
693 static V1_3::ErrorStatus VisitConv2DNode(xnn_subgraph_t subgraph, in VisitConv2DNode()
792 static V1_3::ErrorStatus VisitDepthwiseConv2DNode(xnn_subgraph_t subgraph, in VisitDepthwiseConv2DNode()
893 static V1_3::ErrorStatus VisitDivNode(xnn_subgraph_t subgraph, const V1_3::Operation& operation, in VisitDivNode()
923 static V1_3::ErrorStatus VisitFullyConnectedNode(xnn_subgraph_t subgraph, in VisitFullyConnectedNode()
[all …]
/packages/modules/NeuralNetworks/common/
DModelUtils.cpp144 void identifyUsedPools(const Model::Subgraph& subgraph, std::vector<bool>* used) { in identifyUsedPools()
158 for (const auto& subgraph : model.referenced) { in identifyUsedPools() local
220 for (auto& subgraph : model->referenced) { in fixOperandDataLocations() local
237 const auto collectPrefixes = [collectPrefix](const Model::Subgraph& subgraph) { in identifyUsedExtensions()
243 for (const auto& subgraph : model.referenced) { in identifyUsedExtensions() local
DValidateHal.cpp475 const Model::Subgraph& subgraph = *getSubgraph(modelOperand); in validateOperations() local
481 const Model::Subgraph& subgraph = *getSubgraph(modelOperand); in validateOperations() local
674 static bool checkNoReferenceCycles(const V1_3::Model& model, const V1_3::Subgraph& subgraph, in checkNoReferenceCycles()
730 auto validateSubgraph = [&model, mode](const V1_3::Subgraph& subgraph) -> bool { in validateModel()
DLegacyHalUtils.cpp156 static void logSubgraphToInfo(std::string label, const V1_3::Subgraph& subgraph) { in logSubgraphToInfo()
1564 Model::Subgraph uncheckedConvert(const V1_3::Subgraph& subgraph) { in uncheckedConvert()
1719 V1_3::Subgraph convertToV1_3(const Model::Subgraph& subgraph) { in convertToV1_3()
DCpuExecutor.cpp600 int CpuExecutor::executeSubgraph(const Model::Subgraph& subgraph, RunTimeOperandInfo* operands) { in executeSubgraph()
610 const Model::Subgraph& subgraph) { in initializeRunTimeInfo()
/packages/modules/NeuralNetworks/runtime/
DModelArchHasher.cpp33 bool updateSubgraph(SHA256_CTX* hasher, const Model::Subgraph& subgraph) { in updateSubgraph()
78 for (auto& subgraph : model.referenced) { in calcModelArchHash() local
DModelBuilder.cpp951 Model::Subgraph subgraph; in makeSubgraph() local
960 Model::Subgraph* subgraph) { in updateOperandLocations()
/packages/modules/NeuralNetworks/shim_and_sl/
DShimConverter.cpp58 const auto& subgraph = subgraphIndex == 0 ? model.main : model.referenced[subgraphIndex - 1]; in convertSubgraphFromHAL() local
160 auto subgraph = convertSubgraphFromHAL(nnapi, memoryPools, model, allModels, in convertSubgraphFromHAL() local
261 const auto& subgraph = sindex == 0 ? model.main : model.referenced[sindex - 1]; in needsCopiedOperandValues() local
276 bool isValid(const Subgraph& subgraph) { in isValid()
358 [](const Subgraph& subgraph) { return isValid(subgraph); })); in isValid()
/packages/modules/NeuralNetworks/runtime/test/
DGeneratedTestUtils.cpp116 auto processSubgraph = [&size](const TestSubgraph& subgraph) {
125 for (const TestSubgraph& subgraph : testModel.referenced) { local
135 static void createModelFromSubgraph(const TestSubgraph& subgraph, bool testDynamicOutputShape,
DTestPartitioning.cpp289 uint32_t lookupOperation(const HidlModel& model, const V1_3::Subgraph& subgraph, in lookupOperation()
415 const V1_3::Subgraph& subgraph) { in getSupportedOperationsForSubgraph()
/packages/modules/NeuralNetworks/common/types/src/
DSharedMemory.cpp53 bool hasNoPointerData(const Model::Subgraph& subgraph) { in hasNoPointerData()
76 void copyPointersToSharedMemory(Model::Subgraph* subgraph, ConstantMemoryBuilder* memoryBuilder) { in copyPointersToSharedMemory()
165 [&memoryBuilder](auto& subgraph) { in flushDataFromPointerToShared()
DValidation.cpp782 Result<void> validateExecutionOrder(const Model::Subgraph& subgraph) { in validateExecutionOrder()
837 Result<Version> validateModelSubgraph(const Model::Subgraph& subgraph, in validateModelSubgraph()
935 const auto& subgraph = referencedSubgraphs.at(subgraphIndex); in checkNoReferenceCycles() local
995 const auto& subgraph = model.referenced[referencedIndex]; in validateModel() local
1560 const Model::Subgraph& subgraph = getSubgraph(subgraphs, modelOperand); in getInputOperand() local
1565 const Model::Subgraph& subgraph = getSubgraph(subgraphs, modelOperand); in getOutputOperand() local
DTypeUtils.cpp653 std::ostream& operator<<(std::ostream& os, const Model::Subgraph& subgraph) { in operator <<()
/packages/modules/NeuralNetworks/runtime/test/android_fuzzing/
DFuzzTest.cpp67 const auto& subgraph = index == 0 ? testModel.main : testModel.referenced[index - 1]; in areSubgraphsAcyclic() local
149 auto subgraph = CreateSubgraph(testModel, index, subgraphs); in CreateModels() local
DFuzzHarness.cpp44 bool hasOperandThatOverflows(const TestSubgraph& subgraph) { in hasOperandThatOverflows()
DGenerateCorpus.cpp138 Subgraph convert(const TestSubgraph& subgraph) { in convert()
149 for (const auto& subgraph : subgraphs) { in convert() local
DConverter.cpp155 TestSubgraph convert(const Subgraph& subgraph) { in convert()
176 [](const auto& subgraph) { return convert(subgraph); }); in convert()
DModel.proto218 repeated Subgraph subgraph = 1; field
/packages/modules/NeuralNetworks/tools/test_generator/
Dexample_generator.py131 def GetSubgraphStruct(subgraph): argument
/packages/modules/NeuralNetworks/tools/test_generator/test_harness/
DTestHarness.cpp276 model.forEachSubgraph([&hasFloat16Inputs](const TestSubgraph& subgraph) { in checkResults()
296 auto processSubgraph = [](TestSubgraph* subgraph) { in convertQuant8AsymmOperandsToSigned()
/packages/modules/NeuralNetworks/tools/test_generator/test_harness/include/
DTestHarness.h346 for (const TestSubgraph& subgraph : referenced) { in forEachSubgraph() local