mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
5fbfca3d58
* 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>
37 lines
1.3 KiB
C++
37 lines
1.3 KiB
C++
#include "lib/Api.Experimental/pch/pch.h"
|
|
#include "LearningModelExperimental.h"
|
|
#include "LearningModel.h"
|
|
#include "LearningModelJoinOptions.h"
|
|
|
|
namespace WINML_EXPERIMENTALP {
|
|
|
|
LearningModelExperimental::LearningModelExperimental(Microsoft::AI::MachineLearning::LearningModel const& model) : model_(model)
|
|
{}
|
|
|
|
winml::LearningModel LearningModelExperimental::JoinModel(winml::LearningModel const& other, winml_experimental::LearningModelJoinOptions const& options) {
|
|
telemetry_helper.LogApiUsage("LearningModelExperimental::JoinModel");
|
|
|
|
auto modelp = model_.as<winmlp::LearningModel>();
|
|
auto optionsp = options.as<winml_experimentalp::LearningModelJoinOptions>();
|
|
|
|
modelp->JoinModel(other,
|
|
optionsp->GetLinkages(),
|
|
optionsp->PromoteUnlinkedOutputsToFusedOutputs(),
|
|
optionsp->CloseModelOnJoin(),
|
|
optionsp->JoinedNodePrefix());
|
|
|
|
return model_;
|
|
}
|
|
|
|
void LearningModelExperimental::Save(hstring const& file_name) {
|
|
telemetry_helper.LogApiUsage("LearningModelExperimental::Save");
|
|
auto modelp = model_.as<winmlp::LearningModel>();
|
|
modelp->SaveToFile(file_name);
|
|
}
|
|
|
|
void LearningModelExperimental::SetName(hstring const& model_name) {
|
|
auto modelp = model_.as<winmlp::LearningModel>();
|
|
modelp->SetName(model_name);
|
|
}
|
|
|
|
}
|