onnxruntime/winml/lib/Api.Experimental/LearningModelExperimental.h
Numfor Tiapo 5fbfca3d58
Add Experimental API for setting model name (#10518)
* Add experimental API for editing model name

* Change EditModelName to 'SetName'

* Change API to pass c_string

* Update SetName to edit the proto

* Test that the model proto gets changed

* Remove comments

* Skip inbox tests

* Use filehelper path

Co-authored-by: Numfor Mbiziwo-Tiapo <numform@microsoft.com>
2022-02-25 14:23:49 -08:00

29 lines
881 B
C++

#pragma once
#include "LearningModelExperimental.g.h"
namespace WINML_EXPERIMENTALP {
struct LearningModelExperimental : LearningModelExperimentalT<LearningModelExperimental>
{
LearningModelExperimental() = default;
LearningModelExperimental(Microsoft::AI::MachineLearning::LearningModel const& model);
Microsoft::AI::MachineLearning::LearningModel JoinModel(Microsoft::AI::MachineLearning::LearningModel const& other, Microsoft::AI::MachineLearning::Experimental::LearningModelJoinOptions const& options);
void Save(hstring const& file_name);
void SetName(hstring const& model_name);
private:
Microsoft::AI::MachineLearning::LearningModel model_;
};
}
namespace WINML_EXPERIMENTAL::factory_implementation {
struct LearningModelExperimental : LearningModelExperimentalT<LearningModelExperimental, implementation::LearningModelExperimental>
{
};
}