mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-01 03:45:06 +00:00
6c236fd939
there was a merge conflict in onnxruntime/core/providers/dml/OperatorAuthorHelper/Common.h that required manual merging. Related work items: #36867798
25 lines
956 B
C++
25 lines
956 B
C++
#include "lib/Api.Experimental/pch/pch.h"
|
|
#include "LearningModelOutputs.h"
|
|
#include "LearningModelBuilder.h"
|
|
#include "TensorFeatureDescriptor.h"
|
|
|
|
namespace WINML_EXPERIMENTALP
|
|
{
|
|
|
|
LearningModelOutputs::LearningModelOutputs(winml_experimental::LearningModelBuilder builder) :
|
|
output_descriptors_(winrt::single_threaded_vector<winml::ILearningModelFeatureDescriptor>()),
|
|
builder_(builder) {
|
|
}
|
|
|
|
winml_experimental::LearningModelBuilder LearningModelOutputs::Add(winml::ILearningModelFeatureDescriptor const& output)
|
|
{
|
|
// Perform model update inside the builder
|
|
auto model = builder_.as<winml_experimentalp::LearningModelBuilder>()->UseModel();
|
|
auto descriptor_provider = output.as<_winml::IDescriptorInfoProvider>();
|
|
auto name = _winml::Strings::UTF8FromHString(output.Name());
|
|
model->AddModelOutput(name.c_str(), descriptor_provider.get());
|
|
output_descriptors_.Append(output);
|
|
return builder_;
|
|
}
|
|
|
|
} // namespace WINML_EXPERIMENTALP
|