onnxruntime/winml/test/common/SqueezeNetValidator.h

31 lines
964 B
C
Raw Normal View History

//-----------------------------------------------------------------------------
//
// Copyright (c) Microsoft Corporation. All rights reserved.
//
//-----------------------------------------------------------------------------
#pragma once
#include <std.h>
enum OutputBindingStrategy { Bound, Unbound, Empty };
namespace WinML::Engine::Test::ModelValidator
{
void FnsCandy16(
std::string instance,
winrt::Windows::AI::MachineLearning::LearningModelDeviceKind deviceKind,
OutputBindingStrategy outputBindingStrategy,
bool bindInputsAsIInspectable,
float dataTolerance = false);
void SqueezeNet(
std::string instance,
winrt::Windows::AI::MachineLearning::LearningModelDeviceKind deviceKind,
float dataTolerance,
bool bindAsImage = false,
OutputBindingStrategy outputBindingStrategy = OutputBindingStrategy::Bound,
bool bindInputsAsIInspectable = false
);
}