onnxruntime/winml/test/common/SqueezeNetValidator.h
Ryan Lai 8d017f5701
Scneario Test : Build Google Test and Taef Test based on preprocessor definition (#2809)
* Add winml macro wrappers on top of google test macros

* change test methods to disabled

* Add custom winml macros for both taef and google tests

* PR comments
2020-01-13 09:22:54 -08:00

30 lines
964 B
C++

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