mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-17 21:10:43 +00:00
671fa60327
* switch to work PC * back with iterable of buffers * add raw api tests * tensorization * last test * all tests pass! * small cleanup * whitespace * newline * whitespace * refactor common code into DisjointBufferHelpers * remove unused file * warning * skip gpu tests when hardware not available * Add error condition when createreference is invoked * add null check to cretereference * uncomment out check Co-authored-by: Sheil Kumar <sheilk@microsoft.com>
16 lines
No EOL
544 B
C
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); |