Home
last modified time | relevance | path

Searched refs:Subgraph (Results 1 – 25 of 31) sorted by relevance

12

/packages/modules/NeuralNetworks/runtime/operation_converters/
DSubGraphContext.h34 SubGraphContext(const Model* model, const Model::Subgraph* subgraph,
50 const Model::Subgraph* getSubgraph() const { return mSubgraph; } in getSubgraph()
76 const Model::Subgraph* mSubgraph;
DAddOperationConverter.cpp29 const Model::Subgraph* subgraph = context->getSubgraph(); in convert()
DConv2DOperationConverter.cpp54 const Model::Subgraph* subgraph = context->getSubgraph(); in decomposeExplicitPadding()
145 const Model::Subgraph* subgraph = context->getSubgraph(); in convert()
DDepthwiseConv2DOperationConverter.cpp29 const Model::Subgraph* subgraph = context->getSubgraph(); in convert()
DSubGraphContext.cpp28 SubGraphContext::SubGraphContext(const Model* model, const Model::Subgraph* subgraph, in SubGraphContext()
/packages/modules/NeuralNetworks/common/
DModelUtils.cpp95 void identifyUsedSubgraphs(uint32_t current, const std::vector<Model::Subgraph>& subgraphs,
102 const std::vector<Model::Subgraph>& subgraphs, std::vector<bool>* used) { in identifyUsedSubgraphs()
116 void identifyUsedSubgraphs(uint32_t current, const std::vector<Model::Subgraph>& subgraphs, in identifyUsedSubgraphs()
144 void identifyUsedPools(const Model::Subgraph& subgraph, std::vector<bool>* used) { in identifyUsedPools()
237 const auto collectPrefixes = [collectPrefix](const Model::Subgraph& subgraph) { in identifyUsedExtensions()
DValidateHal.cpp171 const hardware::hidl_vec<V1_3::Subgraph>& subgraphs, in validateOperands()
453 const hardware::hidl_vec<V1_3::Subgraph>& subgraphs, in validateOperations()
464 [&canonicalSubgraphs](const Operand& modelOperand) -> const Model::Subgraph* { in validateOperations()
476 const Model::Subgraph& subgraph = *getSubgraph(modelOperand); in validateOperations()
482 const Model::Subgraph& subgraph = *getSubgraph(modelOperand); in validateOperations()
675 static bool checkNoReferenceCycles(const V1_3::Model& model, const V1_3::Subgraph& subgraph, in checkNoReferenceCycles()
676 std::set<const V1_3::Subgraph*>* path) { in checkNoReferenceCycles()
695 std::set<const V1_3::Subgraph*> path; in checkNoReferenceCycles()
731 auto validateSubgraph = [&model, mode](const V1_3::Subgraph& subgraph) -> bool { in validateModel()
DCpuExecutor.cpp600 int CpuExecutor::executeSubgraph(const Model::Subgraph& subgraph, RunTimeOperandInfo* operands) { in executeSubgraph()
610 const Model::Subgraph& subgraph) { in initializeRunTimeInfo()
647 const_cast<Model::Subgraph*>(&(*mReferencedSubgraphs)[subgraphIndex])); in initializeRunTimeInfo()
1705 const Model::Subgraph& branchSubgraph = in executeIfOperation()
1706 *reinterpret_cast<const Model::Subgraph*>(branchOperand.buffer); in executeIfOperation()
1736 const Model::Subgraph& condSubgraph = in executeWhileOperation()
1737 *reinterpret_cast<const Model::Subgraph*>(condModelOperand.buffer); in executeWhileOperation()
1738 const Model::Subgraph& bodySubgraph = in executeWhileOperation()
1739 *reinterpret_cast<const Model::Subgraph*>(bodyModelOperand.buffer); in executeWhileOperation()
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()
1785 std::vector<Model::Subgraph> uncheckedConvert(const hardware::hidl_vec<V1_3::Subgraph>& subgraphs) { in uncheckedConvert()
1786 return convertVec<Model::Subgraph>(subgraphs); in uncheckedConvert()
/packages/modules/NeuralNetworks/common/types/src/
DValidation.cpp348 Result<Version> validateModelSubgraph(const Model::Subgraph& subgraph,
352 const std::vector<Model::Subgraph>& referenced,
357 const std::vector<Model::Subgraph>& subgraphs, in validateOperandDataLocation()
633 const std::vector<Model::Subgraph>& subgraphs, in validateOperand()
664 const std::vector<size_t>& poolSizes, const std::vector<Model::Subgraph>& subgraphs, in validateOperands()
682 const std::vector<Version>& operandVersions, const std::vector<Model::Subgraph>& subgraphs);
687 const std::vector<Model::Subgraph>& subgraphs) { in validateOperations()
782 Result<void> validateExecutionOrder(const Model::Subgraph& subgraph) { in validateExecutionOrder()
837 Result<Version> validateModelSubgraph(const Model::Subgraph& subgraph, in validateModelSubgraph()
841 const std::vector<Model::Subgraph>& referenced, in validateModelSubgraph()
[all …]
DSharedMemory.cpp36 bool hasNoPointerData(const Model::Subgraph& subgraph);
53 bool hasNoPointerData(const Model::Subgraph& subgraph) { in hasNoPointerData()
76 void copyPointersToSharedMemory(Model::Subgraph* subgraph, ConstantMemoryBuilder* memoryBuilder) { in copyPointersToSharedMemory()
/packages/modules/NeuralNetworks/tools/test_generator/test_harness/
DTestUtils.cpp96 Result<Model::Subgraph> createSubgraph(const TestSubgraph& testSubgraph, in createSubgraph()
116 return Model::Subgraph{.operands = std::move(operands), in createSubgraph()
128 Model::Subgraph mainSubgraph = in createModel()
130 std::vector<Model::Subgraph> refSubgraphs; in createModel()
/packages/modules/NeuralNetworks/common/types/include/nnapi/
DValidation.h120 const std::vector<Model::Subgraph>& subgraphs);
141 const std::vector<size_t>& poolSizes, const std::vector<Model::Subgraph>& subgraphs,
146 const std::vector<Model::Subgraph>& subgraphs,
DTypes.h716 struct Subgraph { struct
767 Subgraph main;
777 std::vector<Subgraph> referenced;
/packages/modules/NeuralNetworks/runtime/test/android_fuzzing/
DModel.proto210 message Subgraph { message
218 repeated Subgraph subgraph = 1;
222 Subgraph main = 1;
DGenerateCorpus.cpp138 Subgraph convert(const TestSubgraph& subgraph) { in convert()
139 Subgraph protoSubgraph; in convert()
/packages/modules/NeuralNetworks/common/include/
DCpuExecutor.h163 std::vector<RunTimeOperandInfo> initializeRunTimeInfo(const Model::Subgraph& subgraph);
171 int executeSubgraph(const Model::Subgraph& subgraph, RunTimeOperandInfo* operands);
184 const std::vector<Model::Subgraph>* mReferencedSubgraphs = nullptr;
DLegacyHalUtils.h346 Model::Subgraph uncheckedConvert(const V1_3::Subgraph& subgraph);
364 std::vector<Model::Subgraph> uncheckedConvert(const hardware::hidl_vec<V1_3::Subgraph>& subgraphs);
401 V1_3::Subgraph convertToV1_3(const Model::Subgraph& model);
/packages/modules/NeuralNetworks/runtime/
DModelBuilder.cpp902 static Model::Subgraph makeSubgraph(const ModelBuilder* model);
906 void updateOperandLocations(const ModelBuilder* refModel, Model::Subgraph* subgraph);
911 std::vector<Model::Subgraph> mRefSubgraphs;
950 Model::Subgraph ModelBuilder::ModelMaker::makeSubgraph(const ModelBuilder* model) { in makeSubgraph()
951 Model::Subgraph subgraph; in makeSubgraph()
960 Model::Subgraph* subgraph) { in updateOperandLocations()
DFlatbufferModelBuilder.h47 Result<SubGraphFlatbuffer> createSubGraphFlatbuffer(const Model::Subgraph& subgraph);
DModelArchHasher.cpp33 bool updateSubgraph(SHA256_CTX* hasher, const Model::Subgraph& subgraph) { in updateSubgraph()
DFlatbufferModelBuilder.cpp84 const Model::Subgraph& subgraph) { in createSubGraphFlatbuffer()
DModelBuilder.h250 std::vector<Model::Subgraph> mReferencedSubgraphsForValidation;
/packages/modules/NeuralNetworks/driver/sample_hidl/
DSampleDriverFloatXNNPACK.cpp109 const V1_3::Subgraph& subgraph, const std::vector<RunTimePoolInfo>& modelPoolInfos, in initializeRunTimeInfo()
154 class Subgraph { class
156 static Subgraph* Create(const hardware::hidl_vec<V1_3::Operation>& operations, in Create()
268 return new Subgraph(runtimePtr, std::move(externals), useStaticBuffer); in Create()
1554 Subgraph(xnn_runtime_t runtime, std::unordered_set<uint32_t>&& externals, in Subgraph() function in android::nn::sample_driver::Subgraph
1614 Subgraph* mSubgraph;
1638 mSubgraph = Subgraph::Create(model->main.operations, mOperands, model->main.inputIndexes, in initialize()
1644 void asyncExecuteXNNPACK(Subgraph* subgraph, RunTimeOperandInfo* operands, in asyncExecuteXNNPACK()
1667 V1_3::ErrorStatus executeXNNPACKBase(Subgraph* subgraph, RunTimeOperandInfo* operands, in executeXNNPACKBase()
1729 executeSynchronouslyXNNPACKBase(Subgraph* subgraph, RunTimeOperandInfo* operands, in executeSynchronouslyXNNPACKBase()
[all …]
/packages/modules/NeuralNetworks/shim_and_sl/
DShimConverter.cpp284 bool isValid(const Subgraph& subgraph) { in isValid()
366 [](const Subgraph& subgraph) { return isValid(subgraph); })); in isValid()

12