Add experimental API for editing model name

winml/api/Microsoft.AI.MachineLearning.Experimental.idl

@@ -141,6 +141,9 @@ namespace ROOT_NS.AI.MachineLearning.Experimental {
 
     //! The JoinModel fuses two models by linking  outputs from the first model, to inupts of the second. 
     ROOT_NS.AI.MachineLearning.LearningModel JoinModel(ROOT_NS.AI.MachineLearning.LearningModel other, LearningModelJoinOptions options);
+
+    //! The EditModelName function changes the model name to the specified string
+    void EditModelName(String model_name);
   }
 
 }  // namespace Microsoft.AI.MachineLearning.Experimental

winml/lib/Api.Experimental/LearningModelExperimental.cpp

@@ -29,4 +29,9 @@ void LearningModelExperimental::Save(hstring const& file_name) {
     modelp->SaveToFile(file_name);
 }
 
+void LearningModelExperimental::EditModelName(hstring const& model_name) {
+  auto modelp = model_.as<winmlp::LearningModel>();
+  modelp->EditModelName(model_name);
+}
+
 }

winml/lib/Api.Experimental/LearningModelExperimental.h

@@ -12,6 +12,8 @@ struct LearningModelExperimental : LearningModelExperimentalT<LearningModelExper
 
     void Save(hstring const& file_name);
 
+    void EditModelName(hstring const& model_name);
+
 private:
     Microsoft::AI::MachineLearning::LearningModel model_;
 };

winml/lib/Api.Ort/OnnxruntimeModel.cpp

@@ -130,6 +130,11 @@ STDMETHODIMP ModelInfo::GetName(const char** out, size_t* len) {
   return S_OK;
 }
 
+STDMETHODIMP ModelInfo::EditModelName(std::string name) {
+  name_ = name;
+  return S_OK;
+}
+
 STDMETHODIMP ModelInfo::GetDomain(const char** out, size_t* len) {
   *out = domain_.c_str();
   *len = domain_.size();

winml/lib/Api.Ort/OnnxruntimeModel.h

@@ -26,6 +26,8 @@ class ModelInfo : public Microsoft::WRL::RuntimeClass<
   (const char** out, size_t* len);
   STDMETHOD(GetName)
   (const char** out, size_t* len);
+  STDMETHOD(EditModelName)
+  (std::string name);
   STDMETHOD(GetDomain)
   (const char** out, size_t* len);
   STDMETHOD(GetDescription)

winml/lib/Api/LearningModel.cpp

@@ -279,6 +279,12 @@ LearningModel::OutputFeatures() try {
 }
 WINML_CATCH_ALL
 
+void LearningModel::EditModelName(const hstring& name) try {
+  auto name_str = _winml::Strings::UTF8FromHString(name);
+  WINML_THROW_IF_FAILED(model_info_->EditModelName(name_str));
+}
+WINML_CATCH_ALL
+
 void LearningModel::Close() try {
   // close the model
   model_ = nullptr;

winml/lib/Api/LearningModel.h

@@ -59,6 +59,8 @@ struct LearningModel : LearningModelT<LearningModel> {
   wfc::IVectorView<winml::ILearningModelFeatureDescriptor>
   OutputFeatures();
 
+  void EditModelName(const hstring& name);
+
   /* IClosable methods. */
   void Close();
 

winml/lib/Common/inc/iengine.h

@@ -66,6 +66,9 @@ IModelInfo : IUnknown {
   STDMETHOD(GetName)
   (const char** out, size_t* len) PURE;
 
+  STDMETHOD(EditModelName)
+  (std::string name) PURE;
+
   STDMETHOD(GetDomain)
   (const char** out, size_t* len) PURE;
 

winml/test/api/LearningModelSessionAPITest.cpp

@@ -1089,6 +1089,20 @@ static void SetIntraOpThreadSpinning() {
     WINML_EXPECT_TRUE(allowSpinning);
  }
 
+ static void EditModelName() {
+   LearningModel model = nullptr;
+   WINML_EXPECT_NO_THROW(APITest::LoadModel(L"model.onnx", model));
+   auto model_name = model.Name();
+   auto squeezenet_old = to_hstring("squeezenet_old");
+   WINML_EXPECT_EQUAL(model_name, squeezenet_old);
+
+   auto experimental_model = winml_experimental::LearningModelExperimental(model);
+   auto new_name = to_hstring("new name");
+   experimental_model.EditModelName(new_name);
+   model_name = model.Name();
+   WINML_EXPECT_EQUAL(model_name, new_name);
+ }
+
 
 const LearningModelSessionAPITestsApi& getapi() {
   static LearningModelSessionAPITestsApi api =
@@ -1123,6 +1137,7 @@ const LearningModelSessionAPITestsApi& getapi() {
     ModelBuilding_STFT,
     ModelBuilding_MelSpectrogramOnThreeToneSignal,
     ModelBuilding_MelWeightMatrix,
+    EditModelName
   };
 
   if (SkipGpuTests()) {

winml/test/api/LearningModelSessionAPITest.h

@@ -34,6 +34,7 @@ struct LearningModelSessionAPITestsApi {
   VoidTest ModelBuilding_STFT;
   VoidTest ModelBuilding_MelSpectrogramOnThreeToneSignal;
   VoidTest ModelBuilding_MelWeightMatrix;
+  VoidTest EditModelName;
 };
 const LearningModelSessionAPITestsApi& getapi();
 
@@ -69,4 +70,5 @@ WINML_TEST(LearningModelSessionAPITests, ModelBuilding_BlackmanWindow)
 WINML_TEST(LearningModelSessionAPITests, ModelBuilding_STFT)
 WINML_TEST(LearningModelSessionAPITests, ModelBuilding_MelSpectrogramOnThreeToneSignal)
 WINML_TEST(LearningModelSessionAPITests, ModelBuilding_MelWeightMatrix)
+WINML_TEST(LearningModelSessionAPITests, EditModelName)
 WINML_TEST_CLASS_END()