Home
last modified time | relevance | path

Searched refs:operationRegistration (Results 1 – 5 of 5) sorted by relevance

/frameworks/ml/nn/common/
DOperationResolver.cpp189 const OperationRegistration* operationRegistration) { in registerOperation() argument
190 CHECK(operationRegistration != nullptr); in registerOperation()
191 auto index = static_cast<int32_t>(operationRegistration->type); in registerOperation()
195 mRegistrations[index] = operationRegistration; in registerOperation()
DCpuExecutor.cpp1703 const OperationRegistration* operationRegistration = in executeOperation() local
1705 if (operationRegistration == nullptr) { in executeOperation()
1707 } else if (operationRegistration->prepare == nullptr || in executeOperation()
1708 operationRegistration->execute == nullptr) { in executeOperation()
1713 success = operationRegistration->flags.allowOmittedOperand || in executeOperation()
1715 success = success && (operationRegistration->flags.allowZeroSizedInput || in executeOperation()
1717 success = success && operationRegistration->prepare(&context) && in executeOperation()
1718 operationRegistration->execute(&context); in executeOperation()
DUtils.cpp1862 const OperationRegistration* operationRegistration = in validateOperation() local
1865 if (operationRegistration == nullptr) { in validateOperation()
1874 if (operationRegistration->validate == nullptr) { in validateOperation()
1878 OperationValidationContext context(operationRegistration->name, inputCount, in validateOperation()
1881 if (!operationRegistration->validate(&context)) { in validateOperation()
/frameworks/ml/nn/runtime/test/fibonacci_extension/
DFibonacciDriver.cpp142 static OperationRegistration operationRegistration(operationType, fibonacci_op::kOperationName, in findOperation() local
148 return prefix != 0 && typeWithinExtension == EXAMPLE_FIBONACCI ? &operationRegistration in findOperation()
/frameworks/ml/nn/common/include/
DOperationResolver.h94 void registerOperation(const OperationRegistration* operationRegistration);