onnxruntime/winml/lib/Api/MapFeatureDescriptor.h
Paul McDaniel 5406801670
Task 23998197: add winml_lib_core into onnnxruntime.dll (#2368)
* Task 23998197: add winml_lib_core into onnnxruntime.dll

* PR feedback
build break on perf_test
2019-11-11 14:34:19 -08:00

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