diff options
Diffstat (limited to 'nn/common/include/nnapi/Validation.h')
-rw-r--r-- | nn/common/include/nnapi/Validation.h | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/nn/common/include/nnapi/Validation.h b/nn/common/include/nnapi/Validation.h index 09133957b..3eda174f6 100644 --- a/nn/common/include/nnapi/Validation.h +++ b/nn/common/include/nnapi/Validation.h @@ -69,10 +69,9 @@ using PreparedModelRole = std::tuple<const IPreparedModel*, IOType, uint32_t>; // IMPORTANT: This function cannot validate dimensions and extraParams with extension operand type. // Each driver should do their own validation of extension type dimensions and extraParams. Result<Version> validateMemoryDesc( - const BufferDesc& desc, - const std::vector<std::shared_ptr<const IPreparedModel>>& preparedModels, + const BufferDesc& desc, const std::vector<SharedPreparedModel>& preparedModels, const std::vector<BufferRole>& inputRoles, const std::vector<BufferRole>& outputRoles, - const std::function<const Model*(const std::shared_ptr<const IPreparedModel>&)>& getModel, + const std::function<const Model*(const SharedPreparedModel&)>& getModel, std::set<PreparedModelRole>* preparedModelRoles, Operand* combinedOperand); Result<void> validateOperandSymmPerChannelQuantParams( |