mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-10 17:37:14 +00:00
27 lines
No EOL
765 B
C++
27 lines
No EOL
765 B
C++
// Copyright (c) Microsoft Corporation.
|
|
// Licensed under the MIT License.
|
|
|
|
#pragma once
|
|
|
|
#include "core/providers/dml/DmlExecutionProvider/src/AbiCustomRegistry.h"
|
|
|
|
namespace Windows::AI::MachineLearning {
|
|
|
|
inline std::list<std::shared_ptr<onnxruntime::CustomRegistry>>
|
|
GetLotusCustomRegistries(
|
|
IMLOperatorRegistry* registry) {
|
|
if (registry != nullptr) {
|
|
// Down-cast to the concrete type.
|
|
// The only supported input is the AbiCustomRegistry type.
|
|
// Other implementations of IMLOperatorRegistry are forbidden.
|
|
auto abi_custom_registry =
|
|
static_cast<winmlp::AbiCustomRegistry*>(registry);
|
|
|
|
// Get the ORT registry
|
|
return abi_custom_registry->GetRegistries();
|
|
}
|
|
|
|
return {};
|
|
}
|
|
|
|
} // namespace Windows::AI::MachineLearning
|