| /packages/modules/NeuralNetworks/runtime/ |
| D | TypeManager.cpp | 214 bool TypeManager::getExtensionPrefix(const std::string& extensionName, uint16_t* prefix) { in getExtensionPrefix() argument
215 auto it = mExtensionNameToPrefix.find(extensionName); in getExtensionPrefix()
221 mExtensionNameToPrefix[extensionName] = *prefix; in getExtensionPrefix()
222 mPrefixToExtension.push_back(&mExtensionNameToExtension[extensionName]); in getExtensionPrefix()
227 bool TypeManager::getExtensionType(const char* extensionName, uint16_t typeWithinExtension, in getExtensionType() argument
230 NN_RET_CHECK(getExtensionPrefix(extensionName, &prefix)); in getExtensionType()
|
| D | NeuralNetworks.cpp | 1450 const char* extensionName, in ANeuralNetworksDevice_getExtensionSupport() argument
1452 if (device == nullptr || extensionName == nullptr || isExtensionSupported == nullptr) { in ANeuralNetworksDevice_getExtensionSupport()
1460 [extensionName](const auto& supportedExtension) { in ANeuralNetworksDevice_getExtensionSupport()
1461 return supportedExtension.name == extensionName; in ANeuralNetworksDevice_getExtensionSupport()
1468 const char* extensionName, in ANeuralNetworksModel_getExtensionOperandType() argument
1472 if (!model || !extensionName || !type) { in ANeuralNetworksModel_getExtensionOperandType()
1477 return m->getExtensionType(extensionName, operandCodeWithinExtension, type); in ANeuralNetworksModel_getExtensionOperandType()
1481 const char* extensionName, in ANeuralNetworksModel_getExtensionOperationType() argument
1485 if (!model || !extensionName || !type) { in ANeuralNetworksModel_getExtensionOperationType()
1490 return m->getExtensionType(extensionName, operationCodeWithinExtension, type); in ANeuralNetworksModel_getExtensionOperationType()
[all …]
|
| D | TypeManager.h | 49 bool getExtensionType(const char* extensionName, uint16_t typeWithinExtension, int32_t* type);
133 bool getExtensionPrefix(const std::string& extensionName, uint16_t* prefix);
|
| D | ModelBuilder.h | 43 int getExtensionType(const char* extensionName, uint16_t typeWithinExtension, int32_t* type);
|
| D | ModelBuilder.cpp | 67 int ModelBuilder::getExtensionType(const char* extensionName, uint16_t typeWithinExtension, in getExtensionType() argument
69 return TypeManager::get()->getExtensionType(extensionName, typeWithinExtension, type) in getExtensionType()
|
| /packages/modules/NeuralNetworks/runtime/include/ |
| D | NeuralNetworksExtensions.h | 52 const char* extensionName, bool* isExtensionSupported)
70 const char* extensionName,
89 const char* extensionName,
|
| D | NeuralNetworksWrapperExtensions.h | 97 int32_t getExtensionOperandType(const char* extensionName, uint16_t typeWithinExtension) { in getExtensionOperandType() argument
99 if (ANeuralNetworksModel_getExtensionOperandType(mModel, extensionName, typeWithinExtension, in getExtensionOperandType()
106 ANeuralNetworksOperationType getExtensionOperationType(const char* extensionName, in getExtensionOperationType() argument
109 if (ANeuralNetworksModel_getExtensionOperationType(mModel, extensionName, in getExtensionOperationType()
|
| /packages/modules/NeuralNetworks/shim_and_sl/public/ |
| D | NeuralNetworksSupportLibraryImpl.h | 253 const char** extensionName);
433 const char* extensionName,
819 const char* extensionName,
830 const char* extensionName,
1016 const char** extensionName);
|
| /packages/modules/NeuralNetworks/shim_and_sl/ |
| D | ShimConverter.cpp | 94 auto* extensionName = getExtensionName(extensionPrefix); in convertSubgraphFromHAL() local
95 if (extensionName == nullptr) { in convertSubgraphFromHAL()
100 resultModel.getExtensionOperandType(*extensionName, typeWithinExtension, in convertSubgraphFromHAL()
200 auto* extensionName = getExtensionName(extensionPrefix); in convertSubgraphFromHAL() local
201 if (extensionName == nullptr) { in convertSubgraphFromHAL()
206 resultModel.getExtensionOperationType(*extensionName, typeWithinExtension, in convertSubgraphFromHAL()
|
| D | ShimDevice.cpp | 158 const char* extensionName; in getVendorExtensions() local
160 &extensionName); in getVendorExtensions()
161 extension.name = extensionName; in getVendorExtensions()
|
| /packages/modules/NeuralNetworks/shim_and_sl/include/ |
| D | SupportLibraryWrapper.h | 296 void getExtensionOperandType(const std::string& extensionName, in getExtensionOperandType() argument
299 mModel, extensionName.c_str(), operandCodeWithinExtension, type) != in getExtensionOperandType()
305 void getExtensionOperationType(const std::string& extensionName, in getExtensionOperationType() argument
309 mModel, extensionName.c_str(), operandCodeWithinExtension, type) != in getExtensionOperationType()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestExtensions.cpp | 111 bool testDriverSupportsExtension(const char* extensionName) { in testDriverSupportsExtension() argument
113 EXPECT_EQ(ANeuralNetworksDevice_getExtensionSupport(mDevice, extensionName, &result), in testDriverSupportsExtension()
|
| D | TestGpuNnapi.cpp | 141 return strcmp(extension.extensionName, requestedExtension) == 0; in isExtensionSupported()
|