UT用例新增

Signed-off-by: 周翔 <zhouxiang78@h-partners.com>
This commit is contained in:
周翔 2024-06-21 14:10:43 +08:00
parent d92876a8d2
commit ac3f4aed02

View File

@ -56,9 +56,9 @@ public:
const ModelConfig&,
std::shared_ptr<PreparedModel>&));
MOCK_METHOD4(PrepareModelFromModelCache, OH_NN_ReturnCode(const std::vector<Buffer>&,
const ModelConfig&,
std::shared_ptr<PreparedModel>&,
bool&));
const ModelConfig&,
std::shared_ptr<PreparedModel>&,
bool&));
MOCK_METHOD3(PrepareOfflineModel, OH_NN_ReturnCode(std::shared_ptr<const mindspore::lite::LiteGraph>,
const ModelConfig&,
std::shared_ptr<PreparedModel>&));