Searched refs:TypeManager (Results 1 – 11 of 11) sorted by relevance
| /frameworks/ml/nn/runtime/test/ |
| D | TestExtensions.cpp | 33 using TypeManager = ::android::nn::TypeManager; typedef
77 TypeManager::get()->forTest_reset(); in SetUp()
84 TypeManager::get()->forTest_reset(); in TearDown()
129 [&](const std::string& binaryPath) -> android::nn::TypeManager::AppPackageInfo { in TEST_F()
138 EXPECT_FALSE(TypeManager::isExtensionsUseAllowed(native_info(""), in TEST_F()
142 EXPECT_FALSE(TypeManager::isExtensionsUseAllowed(native_info("/foobar/foo"), in TEST_F()
146 EXPECT_TRUE(TypeManager::isExtensionsUseAllowed(native_info("/data/foo"), in TEST_F()
150 EXPECT_TRUE(TypeManager::isExtensionsUseAllowed(native_info("/vendor/foo"), in TEST_F()
154 EXPECT_TRUE(TypeManager::isExtensionsUseAllowed(native_info("/odm/foo"), in TEST_F()
158 EXPECT_FALSE(TypeManager::isExtensionsUseAllowed(native_info("/system/foo"), in TEST_F()
[all …]
|
| D | TestValidation.cpp | 138 EXPECT_TRUE(::android::nn::TypeManager::get()->forTest_registerExtension({ in SetUp()
152 ::android::nn::TypeManager::get()->forTest_reset(); in TearDown()
|
| /frameworks/ml/nn/runtime/ |
| D | TypeManager.cpp | 108 bool fetchAppPackageLocationInfo(uid_t uid, TypeManager::AppPackageInfo* appPackageInfo) { in fetchAppPackageLocationInfo()
124 TypeManager::AppPackageInfo appPackageInfo = { in isNNAPIVendorExtensionsUseAllowed()
138 return TypeManager::isExtensionsUseAllowed( in isNNAPIVendorExtensionsUseAllowed()
144 TypeManager::TypeManager() { in TypeManager() function in android::nn::TypeManager
151 bool TypeManager::isExtensionsUseAllowed(const AppPackageInfo& appPackageInfo, in isExtensionsUseAllowed()
184 void TypeManager::findAvailableExtensions() { in findAvailableExtensions()
192 bool TypeManager::registerExtension(Extension extension, const std::string& deviceName) { in registerExtension()
221 bool TypeManager::getExtensionPrefix(const std::string& extensionName, uint16_t* prefix) { in getExtensionPrefix()
234 bool TypeManager::getExtensionType(const char* extensionName, uint16_t typeWithinExtension, in getExtensionType()
242 bool TypeManager::getExtensionInfo(uint16_t prefix, const Extension** extension) const { in getExtensionInfo()
[all …]
|
| D | TypeManager.h | 35 class TypeManager {
37 static TypeManager* get() { in get()
38 static TypeManager manager; in get()
109 void forTest_reset() { *this = TypeManager(); } in forTest_reset()
136 TypeManager();
|
| D | Memory.cpp | 112 const bool isTensor = TypeManager::get()->isTensorType(kOperand.type); in validate()
137 return {.logicalSize = TypeManager::get()->getSizeOfData(kOperand.type, mUpdatedDimensions), in getMetadata()
150 TypeManager::get()->isTensorType(kOperand.type)); in updateMetadata()
155 TypeManager::get()->getSizeOfData(kOperand.type, combined.value())); in updateMetadata()
161 return TypeManager::get()->getSizeOfData(kOperand.type, kInitialDimensions) == 0; in createdWithUnknownShape()
375 if (!TypeManager::get()->isTensorType(operand.type) && !mDesc.dimensions.empty()) { in addRole()
407 if (mOperand.has_value() && !TypeManager::get()->isTensorType(mOperand->type) && in setDimensions()
481 const uint32_t size = TypeManager::get()->getSizeOfData(mOperand->type, mDesc.dimensions); in finish()
504 const uint32_t size = TypeManager::get()->getSizeOfData(mOperand->type, mDesc.dimensions); in allocate()
|
| D | ModelArgumentInfo.cpp | 57 TypeManager::get()->getSizeOfData(operand.type, ret.mDimensions); in createFromPointer()
81 TypeManager::get()->getSizeOfData(operand.type, ret.mDimensions); in createFromMemory()
|
| D | ModelBuilder.cpp | 58 return TypeManager::get()->getExtensionType(extensionName, typeWithinExtension, type) in getExtensionType()
69 if (isExtensionOperandType(operandType) && !TypeManager::get()->areExtensionsAllowed()) { in addOperand()
81 !TypeManager::get()->getExtensionOperandTypeInfo(operandType, &info)) { in addOperand()
129 if (TypeManager::get()->isTensorType(operand.type) && in setOperandValue()
142 uint32_t neededLength = TypeManager::get()->getSizeOfData(operand); in setOperandValue()
325 if (TypeManager::get()->isTensorType(operand.type) && tensorHasUnspecifiedDimensions(operand)) { in setOperandValueFromMemory()
330 uint32_t neededLength = TypeManager::get()->getSizeOfData(operand); in setOperandValueFromMemory()
357 if (isExtensionOperationType(operationType) && !TypeManager::get()->areExtensionsAllowed()) { in addOperation()
981 CHECK(TypeManager::get()->getExtensionInfo(prefix, &extension)); in addExtensionWithPrefix()
|
| D | Android.bp | 54 "TypeManager.cpp",
|
| D | ExecutionBuilder.cpp | 62 NN_RET_CHECK(TypeManager::get()->getExtensionOperandTypeInfo(operand.type, &info)); in checkDimensionInfo()
83 if (!allowUnspecified && TypeManager::get()->isTensorType(operand.type) && in checkDimensionInfo()
908 NN_RET_CHECK(!TypeManager::get()->sizeOfDataOverflowsUInt32(operandType, in updateOutputShapes()
1030 uint32_t length = TypeManager::get()->getSizeOfData(inputOrOutputOperand); in setInputOrOutputFromMemory()
|
| D | Manager.cpp | 673 uint32_t size = TypeManager::get()->getSizeOfData(type, desc.dimensions); in allocate()
|
| /frameworks/ml/nn/runtime/test/fibonacci_extension/ |
| D | FibonacciExtensionTest.cpp | 57 TypeManager::get()->forTest_reset(); in SetUp()
92 TypeManager::get()->forTest_reset(); in TearDown()
|