mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-25 22:26:24 +00:00
* Task 23998197: add winml_lib_core into onnnxruntime.dll * PR feedback build break on perf_test
71 lines
1.8 KiB
C++
71 lines
1.8 KiB
C++
// Copyright (c) Microsoft Corporation.
|
|
// Licensed under the MIT License.
|
|
|
|
#pragma once
|
|
|
|
#include "MapFeatureDescriptor.g.h"
|
|
|
|
namespace winrt::Windows::AI::MachineLearning::implementation {
|
|
struct MapFeatureDescriptor : MapFeatureDescriptorT<
|
|
MapFeatureDescriptor,
|
|
ILearningModelFeatureDescriptorNative> {
|
|
MapFeatureDescriptor() = delete;
|
|
|
|
MapFeatureDescriptor(
|
|
const char* name,
|
|
const char* description,
|
|
bool is_required,
|
|
winml::TensorKind keyKind,
|
|
winml::ILearningModelFeatureDescriptor valueKind);
|
|
|
|
MapFeatureDescriptor(
|
|
hstring const& Name,
|
|
hstring const& Description,
|
|
bool IsRequired,
|
|
Windows::AI::MachineLearning::TensorKind const& KeyKind,
|
|
Windows::AI::MachineLearning::ILearningModelFeatureDescriptor const& ValueDescriptor);
|
|
|
|
// IMapDescriptor
|
|
winml::TensorKind
|
|
KeyKind();
|
|
|
|
winml::ILearningModelFeatureDescriptor
|
|
ValueDescriptor();
|
|
|
|
// IFeatureDescriptor
|
|
hstring
|
|
Name();
|
|
|
|
hstring
|
|
Description();
|
|
|
|
winml::LearningModelFeatureKind
|
|
Kind();
|
|
|
|
bool
|
|
IsRequired();
|
|
|
|
STDMETHOD(GetName)
|
|
(
|
|
const wchar_t** name,
|
|
uint32_t* cchName) override;
|
|
|
|
STDMETHOD(GetDescription)
|
|
(
|
|
const wchar_t** description,
|
|
uint32_t* cchDescription) override;
|
|
|
|
private:
|
|
winrt::hstring name_;
|
|
winrt::hstring description_;
|
|
bool is_required_;
|
|
winml::TensorKind key_kind_;
|
|
winml::ILearningModelFeatureDescriptor value_kind_;
|
|
};
|
|
} // namespace winrt::Windows::AI::MachineLearning::implementation
|
|
|
|
namespace winrt::Windows::AI::MachineLearning::factory_implementation {
|
|
struct MapFeatureDescriptor : MapFeatureDescriptorT<MapFeatureDescriptor, implementation::MapFeatureDescriptor> {
|
|
|
|
};
|
|
} // namespace winrt::Windows::AI::MachineLearning::factory_implementation
|