Searched refs:typeWithinExtension (Results 1 – 8 of 8) sorted by relevance
| /frameworks/ml/nn/runtime/include/ |
| D | NeuralNetworksWrapperExtensions.h | 95 int32_t getExtensionOperandType(const char* extensionName, uint16_t typeWithinExtension) { in getExtensionOperandType() argument
97 if (ANeuralNetworksModel_getExtensionOperandType(mModel, extensionName, typeWithinExtension, in getExtensionOperandType()
105 uint16_t typeWithinExtension) { in getExtensionOperationType() argument
108 typeWithinExtension, in getExtensionOperationType()
|
| /frameworks/ml/nn/runtime/ |
| D | TypeManager.cpp | 252 bool TypeManager::getExtensionType(const char* extensionName, uint16_t typeWithinExtension, in getExtensionType() argument
256 *type = (prefix << kLowBitsType) | typeWithinExtension; in getExtensionType()
271 uint16_t typeWithinExtension = operandType & ((1 << kLowBitsType) - 1); in getExtensionOperandTypeInfo() local
276 extension->operandTypes.begin(), extension->operandTypes.end(), typeWithinExtension, in getExtensionOperandTypeInfo()
281 static_cast<uint16_t>(it->type) == typeWithinExtension) in getExtensionOperandTypeInfo()
282 << "Cannot find operand type " << typeWithinExtension << " in extension " in getExtensionOperandTypeInfo()
|
| D | TypeManager.h | 45 bool getExtensionType(const char* extensionName, uint16_t typeWithinExtension, int32_t* type);
|
| D | ModelBuilder.h | 40 int getExtensionType(const char* extensionName, uint16_t typeWithinExtension, int32_t* type);
|
| D | ModelBuilder.cpp | 50 int ModelBuilder::getExtensionType(const char* extensionName, uint16_t typeWithinExtension, in getExtensionType() argument
52 return TypeManager::get()->getExtensionType(extensionName, typeWithinExtension, type) in getExtensionType()
|
| /frameworks/ml/nn/common/ |
| D | UtilsTest.cpp | 57 static int32_t getExtensionType(uint16_t extensionPrefix, uint16_t typeWithinExtension) { in getExtensionType() argument
60 int32_t type = (extensionPrefix << kLowBitsType) | typeWithinExtension; in getExtensionType()
|
| /frameworks/ml/nn/runtime/test/fibonacci_extension/ |
| D | FibonacciDriver.cpp | 143 uint16_t typeWithinExtension = static_cast<int32_t>(operationType) & kTypeWithinExtensionMask; in findOperation() local
145 return prefix != 0 && typeWithinExtension == TEST_VENDOR_FIBONACCI ? &operationRegistration in findOperation()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidation.cpp | 134 int32_t getExtensionOperandType(uint16_t typeWithinExtension) { in getExtensionOperandType() argument
137 typeWithinExtension, &result), in getExtensionOperandType()
|