mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
f46956056d
Add WinML Experimental API to Register ORT CustomOps Libraries --------- Co-authored-by: Sheil Kumar <sheilk@microsoft.com>
28 lines
949 B
C++
28 lines
949 B
C++
#pragma once
|
|
|
|
#include "LearningModelSessionOptionsExperimental.g.h"
|
|
|
|
namespace WINML_EXPERIMENTALP {
|
|
|
|
struct LearningModelSessionOptionsExperimental : LearningModelSessionOptionsExperimentalT<LearningModelSessionOptionsExperimental>
|
|
{
|
|
LearningModelSessionOptionsExperimental(const winml::LearningModelSessionOptions& options);
|
|
LearningModelSessionOptionsExperimental(const winml::LearningModelSession& session);
|
|
|
|
wfc::IMapView<winrt::hstring, uint32_t> GetNamedDimensionOverrides();
|
|
void RegisterCustomOpsLibrary(const hstring& path);
|
|
|
|
private:
|
|
wfc::IMapView<winrt::hstring, uint32_t> overrides_;
|
|
winml::LearningModelSessionOptions options_;
|
|
};
|
|
|
|
} // namespace WINML_EXPERIMENTALP
|
|
|
|
namespace WINML_EXPERIMENTAL::factory_implementation {
|
|
|
|
struct LearningModelSessionOptionsExperimental : LearningModelSessionOptionsExperimentalT<LearningModelSessionOptionsExperimental, implementation::LearningModelSessionOptionsExperimental>
|
|
{
|
|
};
|
|
|
|
}
|