onnxruntime/winml/test/api/RawApiHelpers.h

21 lines
547 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
);