Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/
DOperationResolver.cpp165 const OperationRegistration* operationRegistration) { in registerOperation() argument
166 CHECK(operationRegistration != nullptr); in registerOperation()
167 auto index = static_cast<int32_t>(operationRegistration->type); in registerOperation()
171 mRegistrations[index] = operationRegistration; in registerOperation()
DCpuExecutor.cpp1730 const OperationRegistration* operationRegistration = in executeOperation() local
1732 if (operationRegistration == nullptr) { in executeOperation()
1734 } else if (operationRegistration->prepare == nullptr || in executeOperation()
1735 operationRegistration->execute == nullptr) { in executeOperation()
1740 success = operationRegistration->flags.allowOmittedOperand || in executeOperation()
1742 success = success && (operationRegistration->flags.allowZeroSizedInput || in executeOperation()
1744 success = success && operationRegistration->prepare(&context) && in executeOperation()
1745 operationRegistration->execute(&context); in executeOperation()
DUtils.cpp1718 const OperationRegistration* operationRegistration = in validateOperation() local
1721 if (operationRegistration == nullptr) { in validateOperation()
1730 if (operationRegistration->validate == nullptr) { in validateOperation()
1736 if (!operationRegistration->validate(&context)) { in validateOperation()
/frameworks/ml/nn/runtime/test/fibonacci_extension/
DFibonacciDriver.cpp139 static OperationRegistration operationRegistration(operationType, fibonacci_op::kOperationName, in findOperation() local
145 return prefix != 0 && typeWithinExtension == TEST_VENDOR_FIBONACCI ? &operationRegistration in findOperation()
/frameworks/ml/nn/common/include/
DOperationResolver.h94 void registerOperation(const OperationRegistration* operationRegistration);