Searched refs:operationRegistration (Results 1 – 6 of 6) sorted by relevance
187 const OperationRegistration* operationRegistration) { in registerOperation() argument188 CHECK(operationRegistration != nullptr); in registerOperation()189 auto index = static_cast<int32_t>(operationRegistration->type); in registerOperation()193 mRegistrations[index] = operationRegistration; in registerOperation()
1648 const OperationRegistration* operationRegistration = in executeOperation() local1650 if (operationRegistration == nullptr) { in executeOperation()1652 } else if (operationRegistration->prepare == nullptr || in executeOperation()1653 operationRegistration->execute == nullptr) { in executeOperation()1657 success = operationRegistration->flags.allowOmittedOperand || in executeOperation()1659 success = success && (operationRegistration->flags.allowZeroSizedInput || in executeOperation()1661 success = success && operationRegistration->prepare(&context) && in executeOperation()1662 operationRegistration->execute(&context); in executeOperation()
1707 const OperationRegistration* operationRegistration = in validateOperation() local1710 if (operationRegistration == nullptr) { in validateOperation()1719 if (operationRegistration->validate == nullptr) { in validateOperation()1723 OperationValidationContext context(operationRegistration->name, inputCount, in validateOperation()1726 const auto maybeVersion = operationRegistration->validate(&context); in validateOperation()
2667 const OperationRegistration* operationRegistration = in validateOperationButNotOperandsImpl() local2671 NN_VALIDATE(operationRegistration != nullptr) << opType << " not registered"; in validateOperationButNotOperandsImpl()2673 NN_VALIDATE(operationRegistration->validate != nullptr) in validateOperationButNotOperandsImpl()2676 OperationValidationContext context(operationRegistration->name, inputIndexes, in validateOperationButNotOperandsImpl()2678 return operationRegistration->validate(&context); in validateOperationButNotOperandsImpl()
141 static OperationRegistration operationRegistration(operationType, fibonacci_op::kOperationName, in findOperation() local147 return prefix != 0 && typeWithinExtension == EXAMPLE_FIBONACCI ? &operationRegistration in findOperation()
99 void registerOperation(const OperationRegistration* operationRegistration);