mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-15 20:50:42 +00:00
eeef157888
winml/ was previously excluded from lintrunner config. This change includes the directory and adds the clang-format config file specific to winml/ that fits existing style. --------- Signed-off-by: Justin Chu <justinchu@microsoft.com>
31 lines
1.2 KiB
C++
31 lines
1.2 KiB
C++
#pragma once
|
|
|
|
#include "LearningModelInputs.g.h"
|
|
#include "LearningModelBuilder.h"
|
|
|
|
namespace WINML_EXPERIMENTALP {
|
|
|
|
struct LearningModelInputs : LearningModelInputsT<LearningModelInputs> {
|
|
LearningModelInputs(winml_experimental::LearningModelBuilder builder);
|
|
|
|
winml_experimental::LearningModelBuilder Add(winml::ILearningModelFeatureDescriptor const& input);
|
|
winml_experimental::LearningModelBuilder Add(
|
|
hstring const& input_name, hstring const& input_description, Windows::Foundation::IInspectable const& default_value
|
|
);
|
|
winml_experimental::LearningModelBuilder AddConstant(
|
|
hstring const& input_name, Windows::Foundation::IInspectable const& value
|
|
);
|
|
winml_experimental::LearningModelBuilder AddInput(
|
|
winml::ILearningModelFeatureDescriptor const& input,
|
|
Windows::Foundation::IInspectable const& default_value,
|
|
bool is_constant
|
|
);
|
|
|
|
private:
|
|
wfc::IVector<winml::ILearningModelFeatureDescriptor> input_descriptors_;
|
|
wfc::IVector<wf::IInspectable> input_default_values_;
|
|
wfc::IVector<winml::ILearningModelFeatureDescriptor> constant_descriptors_;
|
|
wfc::IVector<wf::IInspectable> constant_values_;
|
|
winml_experimental::LearningModelBuilder builder_;
|
|
};
|
|
} // namespace WINML_EXPERIMENTALP
|