diff options
Diffstat (limited to 'nn/common/include/OperationResolver.h')
-rw-r--r-- | nn/common/include/OperationResolver.h | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/nn/common/include/OperationResolver.h b/nn/common/include/OperationResolver.h index d2c066cd3..155341a1a 100644 --- a/nn/common/include/OperationResolver.h +++ b/nn/common/include/OperationResolver.h @@ -32,7 +32,7 @@ struct OperationRegistration { const char* name; // Validates operand types, shapes, and any values known during graph creation. - std::function<bool(const IOperationValidationContext*)> validate; + std::function<Result<Version>(const IOperationValidationContext*)> validate; // prepare is called when the inputs this operation depends on have been // computed. Typically, prepare does any remaining validation and sets @@ -50,10 +50,11 @@ struct OperationRegistration { bool allowZeroSizedInput = false; } flags; - OperationRegistration(OperationType type, const char* name, - std::function<bool(const IOperationValidationContext*)> validate, - std::function<bool(IOperationExecutionContext*)> prepare, - std::function<bool(IOperationExecutionContext*)> execute, Flag flags) + OperationRegistration( + OperationType type, const char* name, + std::function<Result<Version>(const IOperationValidationContext*)> validate, + std::function<bool(IOperationExecutionContext*)> prepare, + std::function<bool(IOperationExecutionContext*)> execute, Flag flags) : type(type), name(name), validate(std::move(validate)), |