mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-25 22:26:24 +00:00
21 lines
580 B
C
21 lines
580 B
C
|
// Copyright (c) Microsoft Corporation.
|
||
|
|
// Licensed under the MIT License.
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
#include "pch.h"
|
||
|
|
|
||
|
|
namespace Windows::AI::MachineLearning {
|
||
|
|
|
||
|
|
struct FeatureDescriptorFactory {
|
||
|
|
FeatureDescriptorFactory(
|
||
|
|
const std::unordered_map<std::string, std::string>& model_metadata);
|
||
|
|
|
||
|
|
wfc::IVector<winml::ILearningModelFeatureDescriptor>
|
||
|
|
CreateDescriptorsFromValueInfoProtos(
|
||
|
|
const std::vector<const onnx::ValueInfoProto*>& value_info_protos);
|
||
|
|
|
||
|
|
private:
|
||
|
|
const std::unordered_map<std::string, std::string>& metadata_;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace Windows::AI::MachineLearning
|