mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-25 22:26:24 +00:00
23 lines
No EOL
692 B
C++
23 lines
No EOL
692 B
C++
// Copyright (c) Microsoft Corporation.
|
|
// Licensed under the MIT License.
|
|
|
|
#include "pch.h"
|
|
|
|
#include "CpuOrtSessionBuilder.h"
|
|
#include "DmlOrtSessionBuilder.h"
|
|
|
|
#include "LearningModelDevice.h"
|
|
|
|
using namespace Windows::AI::MachineLearning;
|
|
|
|
std::unique_ptr<IOrtSessionBuilder> Windows::AI::MachineLearning::CreateOrtSessionBuilder(
|
|
winml::LearningModelDevice const& device) {
|
|
auto device_impl = device.as<winmlp::LearningModelDevice>();
|
|
|
|
auto session_builder =
|
|
device_impl->IsCpuDevice()
|
|
? std::unique_ptr<IOrtSessionBuilder>(new CpuOrtSessionBuilder())
|
|
: std::unique_ptr<IOrtSessionBuilder>(new DmlOrtSessionBuilder(device));
|
|
|
|
return session_builder;
|
|
} |