onnxruntime/winml/test/api/RawApiHelpers.h
Dwayne Robinson 502f85c883 Merged PR 5411855: Manual pull from onnxruntime upstream GitHub master into DmlDev
I'm just doing this manually this once because some commits were squashed, and I want to be sure they merge cleanly for the nightly automatic merge. "git merge upstream/master"
2020-11-17 04:20:47 +00:00

16 lines
No EOL
544 B
C

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#include "microsoft.ai.machinelearning.h"
#include "microsoft.ai.machinelearning.native.h"
#include "raw/microsoft.ai.machinelearning.h"
#include "raw/microsoft.ai.machinelearning.gpu.h"
enum class InputStrategy {
CopyInputs,
BindAsReference,
BindWithMultipleReferences
};
void RunOnDevice(Microsoft::AI::MachineLearning::learning_model& model, Microsoft::AI::MachineLearning::learning_model_device& device, InputStrategy strategy);