|
|
|
@ -118,7 +118,6 @@ Status ModelImpl::Build() {
|
|
|
|
|
MS_LOG(ERROR) << "Build model failed.";
|
|
|
|
|
return static_cast<StatusCode>(ret);
|
|
|
|
|
}
|
|
|
|
|
session->BindThread(true);
|
|
|
|
|
session_.swap(session);
|
|
|
|
|
model->Free();
|
|
|
|
|
MS_LOG(DEBUG) << "Build model success.";
|
|
|
|
@ -180,7 +179,9 @@ Status ModelImpl::Predict(const std::vector<MSTensor> &inputs, std::vector<MSTen
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
session_->BindThread(true);
|
|
|
|
|
auto ret = session_->RunGraph();
|
|
|
|
|
session_->BindThread(false);
|
|
|
|
|
ResetTensorData(old_data, input_tensors);
|
|
|
|
|
if (ret != RET_OK) {
|
|
|
|
|
MS_LOG(ERROR) << "Run graph failed.";
|
|
|
|
|