| /packages/modules/NeuralNetworks/driver/sample/ |
| D | SampleDriverFloatXNNPACK.cpp | 109 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/shim_and_sl/ |
| D | ShimConverter.cpp | 58 const auto& subgraph = subgraphIndex == 0 ? model.main : model.referenced[subgraphIndex - 1]; in convertSubgraphFromHAL() local
154 auto subgraph = convertSubgraphFromHAL(nnapi, memoryPools, model, allModels, in convertSubgraphFromHAL() local
255 const auto& subgraph = sindex == 0 ? model.main : model.referenced[sindex - 1]; in needsCopiedOperandValues() local
270 bool isValid(const Subgraph& subgraph) { in isValid()
352 [](const Subgraph& subgraph) { return isValid(subgraph); })); in isValid()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | GeneratedTestUtils.cpp | 114 auto processSubgraph = [&size](const TestSubgraph& subgraph) {
123 for (const TestSubgraph& subgraph : testModel.referenced) { local
133 static void createModelFromSubgraph(const TestSubgraph& subgraph, bool testDynamicOutputShape,
|
| D | TestPartitioning.cpp | 288 uint32_t lookupOperation(const HidlModel& model, const V1_3::Subgraph& subgraph, in lookupOperation()
414 const V1_3::Subgraph& subgraph) { in getSupportedOperationsForSubgraph()
|
| /packages/modules/NeuralNetworks/tools/test_generator/ |
| D | example_generator.py | 128 def GetSubgraphStruct(subgraph): argument
|
| /packages/modules/NeuralNetworks/common/ |
| D | ValidateHal.cpp | 475 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()
|
| D | Validation.cpp | 809 Result<void> validateExecutionOrder(const Model::Subgraph& subgraph) { in validateExecutionOrder()
862 Result<Version> validateModelSubgraph(const Model::Subgraph& subgraph, in validateModelSubgraph()
959 const auto& subgraph = referencedSubgraphs.at(subgraphIndex); in checkNoReferenceCycles() local
1019 const auto& subgraph = model.referenced[referencedIndex]; in validateModel() local
1606 const Model::Subgraph& subgraph = getSubgraph(subgraphs, modelOperand); in getInputOperand() local
1611 const Model::Subgraph& subgraph = getSubgraph(subgraphs, modelOperand); in getOutputOperand() local
|
| D | LegacyHalUtils.cpp | 155 static void logSubgraphToInfo(std::string label, const V1_3::Subgraph& subgraph) { in logSubgraphToInfo()
1561 Model::Subgraph uncheckedConvert(const V1_3::Subgraph& subgraph) { in uncheckedConvert()
1716 V1_3::Subgraph convertToV1_3(const Model::Subgraph& subgraph) { in convertToV1_3()
|
| D | CpuExecutor.cpp | 600 int CpuExecutor::executeSubgraph(const Model::Subgraph& subgraph, RunTimeOperandInfo* operands) { in executeSubgraph()
610 const Model::Subgraph& subgraph) { in initializeRunTimeInfo()
|
| D | TypeUtils.cpp | 775 std::ostream& operator<<(std::ostream& os, const Model::Subgraph& subgraph) { in operator <<()
|
| /packages/modules/NeuralNetworks/runtime/ |
| D | ModelBuilder.cpp | 913 Model::Subgraph subgraph; in makeSubgraph() local
922 Model::Subgraph* subgraph) { in updateOperandLocations()
|
| /packages/modules/NeuralNetworks/tools/test_generator/test_harness/ |
| D | TestHarness.cpp | 276 model.forEachSubgraph([&hasFloat16Inputs](const TestSubgraph& subgraph) { in checkResults()
296 auto processSubgraph = [](TestSubgraph* subgraph) { in convertQuant8AsymmOperandsToSigned()
|
| /packages/modules/NeuralNetworks/tools/test_generator/test_harness/include/ |
| D | TestHarness.h | 319 for (const TestSubgraph& subgraph : referenced) { in forEachSubgraph() local
|