onnxruntime/winml/test/api/RawApiHelpers.h

16 lines
544 B
C
Raw Normal View History

// 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);