add GetInputNames GetOutputNames

雾聪

2024-10-10 1480dcf5d571c4920b4f18717d580646794b8d28

 runtime/onnxruntime/src/sensevoice-small.cpp

@@ -42,26 +42,8 @@
        exit(-1);
    }

    string strName;
    GetInputName(m_session_.get(), strName);
    m_strInputNames.push_back(strName.c_str());
    GetInputName(m_session_.get(), strName,1);
    m_strInputNames.push_back(strName);
    GetInputName(m_session_.get(), strName,2);
    m_strInputNames.push_back(strName);
    GetInputName(m_session_.get(), strName,3);
    m_strInputNames.push_back(strName);

    size_t numOutputNodes = m_session_->GetOutputCount();
    for(int index=0; index<numOutputNodes; index++){
        GetOutputName(m_session_.get(), strName, index);
        m_strOutputNames.push_back(strName);
    }

    for (auto& item : m_strInputNames)
        m_szInputNames.push_back(item.c_str());
    for (auto& item : m_strOutputNames)
        m_szOutputNames.push_back(item.c_str());
    GetInputNames(m_session_.get(), m_strInputNames, m_szInputNames);
    GetOutputNames(m_session_.get(), m_strOutputNames, m_szOutputNames);
    vocab = new Vocab(token_file.c_str());
    LoadCmvn(am_cmvn.c_str());
}