mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-14 20:48:00 +00:00
2717c178cc
* Migrate winml to Microsoft Namespace (packaging changes are pending) * add ns_prefix toggle * fix packaging * Users/sheilk/add missing raw header (#3484) * add dualapipartition * wrong variable for repo root Co-authored-by: Sheil Kumar <sheilk@microsoft.com> * remove existence check to force failures * extra paren * dualapipartition needs to be referenced from the source * add microsoft.ai.machinelearning.dll to the output dir * rename the idl file so that assembly info is correctly added into the winmd * fix namespaces * update namespaces * default to microsoft, and add namespace override as build argument * update cmakesetings.json as well * remove from cmakelists.txt Co-authored-by: Sheil Kumar <sheilk@microsoft.com> Co-authored-by: Changming Sun <chasun@microsoft.com>
86 lines
2.2 KiB
C++
86 lines
2.2 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#include "pch.h"
|
|
|
|
#include "LearningModel.h"
|
|
|
|
#include "TensorFeatureDescriptor.h"
|
|
|
|
namespace WINMLP {
|
|
TensorFeatureDescriptor::TensorFeatureDescriptor(
|
|
const char* name,
|
|
const char* description,
|
|
winml::TensorKind tensor_kind,
|
|
const std::vector<int64_t>& shape,
|
|
bool is_required,
|
|
bool has_unsupported_image_metadata) : name_(_winml::Strings::HStringFromUTF8(name)),
|
|
description_(_winml::Strings::HStringFromUTF8(description)),
|
|
tensor_kind_(tensor_kind),
|
|
shape_(shape),
|
|
is_required_(is_required),
|
|
has_unsupported_image_metadata_(has_unsupported_image_metadata) {
|
|
}
|
|
|
|
winml::TensorKind
|
|
TensorFeatureDescriptor::TensorKind() try {
|
|
return tensor_kind_;
|
|
}
|
|
WINML_CATCH_ALL
|
|
|
|
wfc::IVectorView<int64_t>
|
|
TensorFeatureDescriptor::Shape() try {
|
|
return winrt::single_threaded_vector<int64_t>(
|
|
std::vector<int64_t>(
|
|
std::begin(shape_),
|
|
std::end(shape_)))
|
|
.GetView();
|
|
}
|
|
WINML_CATCH_ALL
|
|
|
|
winrt::hstring
|
|
TensorFeatureDescriptor::Name() try {
|
|
return name_;
|
|
}
|
|
WINML_CATCH_ALL
|
|
|
|
winrt::hstring
|
|
TensorFeatureDescriptor::Description() try {
|
|
return description_;
|
|
}
|
|
WINML_CATCH_ALL
|
|
|
|
winml::LearningModelFeatureKind
|
|
TensorFeatureDescriptor::Kind() try {
|
|
return LearningModelFeatureKind::Tensor;
|
|
}
|
|
WINML_CATCH_ALL
|
|
|
|
bool TensorFeatureDescriptor::IsRequired() try {
|
|
return is_required_;
|
|
}
|
|
WINML_CATCH_ALL
|
|
|
|
bool TensorFeatureDescriptor::IsUnsupportedMetaData() try {
|
|
return has_unsupported_image_metadata_;
|
|
}
|
|
WINML_CATCH_ALL
|
|
|
|
HRESULT
|
|
TensorFeatureDescriptor::GetName(
|
|
const wchar_t** name,
|
|
uint32_t* cchName) {
|
|
*name = name_.data();
|
|
*cchName = static_cast<uint32_t>(name_.size());
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT
|
|
TensorFeatureDescriptor::GetDescription(
|
|
const wchar_t** description,
|
|
uint32_t* cchDescription) {
|
|
*description = description_.data();
|
|
*cchDescription = static_cast<uint32_t>(description_.size());
|
|
return S_OK;
|
|
}
|
|
} // namespace WINMLP
|