mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
82 lines
No EOL
2 KiB
C++
82 lines
No EOL
2 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#include "pch.h"
|
|
|
|
#include "MapFeatureDescriptor.h"
|
|
|
|
namespace WINMLP {
|
|
MapFeatureDescriptor::MapFeatureDescriptor(
|
|
const char* name,
|
|
const char* description,
|
|
bool is_required,
|
|
winml::TensorKind key_kind,
|
|
winml::ILearningModelFeatureDescriptor value_kind) : name_(_winml::Strings::HStringFromUTF8(name)),
|
|
description_(_winml::Strings::HStringFromUTF8(description)),
|
|
is_required_(is_required),
|
|
key_kind_(key_kind),
|
|
value_kind_(value_kind) {
|
|
}
|
|
|
|
winml::TensorKind
|
|
MapFeatureDescriptor::KeyKind() try {
|
|
return key_kind_;
|
|
}
|
|
WINML_CATCH_ALL
|
|
|
|
winml::ILearningModelFeatureDescriptor
|
|
MapFeatureDescriptor::ValueDescriptor() try {
|
|
return value_kind_;
|
|
}
|
|
WINML_CATCH_ALL
|
|
|
|
hstring
|
|
MapFeatureDescriptor::Name() try {
|
|
return name_;
|
|
}
|
|
WINML_CATCH_ALL
|
|
|
|
hstring
|
|
MapFeatureDescriptor::Description() try {
|
|
return description_;
|
|
}
|
|
WINML_CATCH_ALL
|
|
|
|
winml::LearningModelFeatureKind
|
|
MapFeatureDescriptor::Kind() try {
|
|
return LearningModelFeatureKind::Map;
|
|
}
|
|
WINML_CATCH_ALL
|
|
|
|
bool MapFeatureDescriptor::IsRequired() try {
|
|
return is_required_;
|
|
}
|
|
WINML_CATCH_ALL
|
|
|
|
HRESULT
|
|
MapFeatureDescriptor::GetName(
|
|
const wchar_t** name,
|
|
uint32_t* cchName) {
|
|
*name = name_.data();
|
|
*cchName = static_cast<uint32_t>(name_.size());
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT
|
|
MapFeatureDescriptor::GetDescription(
|
|
const wchar_t** description,
|
|
uint32_t* cchDescription) {
|
|
*description = description_.data();
|
|
*cchDescription = static_cast<uint32_t>(description_.size());
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT
|
|
MapFeatureDescriptor::GetDescriptorInfo(
|
|
_winml::IEngineFactory* engine_factory,
|
|
_winml::IDescriptorInfo** info) {
|
|
engine_factory->CreateMapDescriptorInfo(info);
|
|
return S_OK;
|
|
}
|
|
|
|
} // namespace WINMLP
|