mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-26 22:35:43 +00:00
21 lines
No EOL
580 B
C++
21 lines
No EOL
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
|