|
|
|
@ -49,6 +49,7 @@ class MS_API Model {
|
|
|
|
|
std::vector<MSTensor> GetOutputs();
|
|
|
|
|
inline std::vector<std::string> GetOutputTensorNames();
|
|
|
|
|
inline MSTensor GetOutputByTensorName(const std::string &tensor_name);
|
|
|
|
|
inline std::vector<MSTensor> GetOutputsByNodeName(const std::string &tensor_name);
|
|
|
|
|
|
|
|
|
|
static bool CheckModelSupport(enum DeviceType device_type, ModelType model_type);
|
|
|
|
|
|
|
|
|
@ -71,5 +72,9 @@ std::vector<std::string> Model::GetOutputTensorNames() { return VectorCharToStri
|
|
|
|
|
MSTensor Model::GetOutputByTensorName(const std::string &tensor_name) {
|
|
|
|
|
return GetOutputByTensorName(StringToChar(tensor_name));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
std::vector<MSTensor> Model::GetOutputsByNodeName(const std::string &tensor_name) {
|
|
|
|
|
return GetOutputsByNodeName(StringToChar(tensor_name));
|
|
|
|
|
}
|
|
|
|
|
} // namespace mindspore
|
|
|
|
|
#endif // MINDSPORE_INCLUDE_API_MODEL_H
|
|
|
|
|