mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-14 20:48:00 +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>
46 lines
1.7 KiB
C++
46 lines
1.7 KiB
C++
#pragma once
|
|
|
|
#include <winrt/Windows.Foundation.Collections.h>
|
|
#include "LearningModelBuilder.g.h"
|
|
#include "iengine.h"
|
|
|
|
namespace WINML_EXPERIMENTALP {
|
|
|
|
struct LearningModelBuilder : LearningModelBuilderT<LearningModelBuilder> {
|
|
LearningModelBuilder(int64_t opset);
|
|
LearningModelBuilder(LearningModelBuilder& builder);
|
|
|
|
winml_experimental::LearningModelInputs Inputs();
|
|
winml_experimental::LearningModelOutputs Outputs();
|
|
winml_experimental::LearningModelOperatorSet Operators();
|
|
winml::LearningModel CreateModel();
|
|
void Save(const winrt::hstring& file_name);
|
|
|
|
static winml_experimental::LearningModelBuilder Create(int32_t opset);
|
|
|
|
static winml::TensorFeatureDescriptor CreateTensorFeatureDescriptor(
|
|
hstring const& name, hstring const& description, winml::TensorKind const& kind, array_view<int64_t const> shape
|
|
);
|
|
|
|
static winml::TensorFeatureDescriptor CreateTensorFeatureDescriptor(
|
|
hstring const& name, winml::TensorKind const& kind, array_view<int64_t const> shape
|
|
);
|
|
|
|
_winml::IModel* UseModel();
|
|
|
|
winml::LearningModelSession InertSession() { return inert_session_; }
|
|
|
|
private:
|
|
com_ptr<_winml::IEngineFactory> engine_factory_;
|
|
winml::LearningModelSession inert_session_;
|
|
com_ptr<_winml::IModel> model_;
|
|
|
|
winml_experimental::LearningModelInputs inputs_;
|
|
winml_experimental::LearningModelOutputs outputs_;
|
|
winml_experimental::LearningModelOperatorSet operators_;
|
|
};
|
|
} // namespace WINML_EXPERIMENTALP
|
|
|
|
namespace WINML_EXPERIMENTAL::factory_implementation {
|
|
struct LearningModelBuilder : LearningModelBuilderT<LearningModelBuilder, implementation::LearningModelBuilder> {};
|
|
} // namespace WINML_EXPERIMENTAL::factory_implementation
|