colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Set Graph Version during in Compile() 

Signed-off-by: Kainan Cha <kainan.zha@verisilicon.com>
This commit is contained in:
Kainan Cha 2021-03-29 17:11:06 +08:00
parent c569555f1f
commit f92a5de68b
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/tim/vx/graph.cc
View File

@ -74,6 +74,12 @@ std::shared_ptr<Tensor> GraphImpl::CreateTensorPlaceHolder() {
bool GraphImpl::Compile() {
bool status = true;
auto major = vsi_nn_GetVersionMajor();
auto minor = vsi_nn_GetVersionMinor();
auto patch = vsi_nn_GetVersionPatch();
vsi_nn_SetGraphVersion(graph_,major,minor,patch);
std::call_once(setio_once_, [&status, this]() {
status = (vsi_nn_SetGraphInputs(this->graph_, this->inputs_.data(), this->inputs_.size()) &&
vsi_nn_SetGraphOutputs(this->graph_, this->outputs_.data(), this->outputs_.size()));