onnxruntime/winml/test/api/LearningModelBindingAPITest.h
Sheil Kumar b72fe13941
Update WinML Projection to accept sequence of tensors (#3287)
* Enable sequence of tensor

* add tests

* small updates

* There should only be 2 elements returned

* CR feedback, and another 6->2 check update in the test.

* missing semicolon...

* Add explicit to constructor taking pointer paramter

Co-authored-by: Sheil Kumar <sheilk@microsoft.com>
2020-03-23 15:55:20 -07:00

56 lines
No EOL
2.6 KiB
C

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#include "test.h"
struct LearningModelBindingAPITestApi {
SetupTest LearningModelBindingAPITestSetup;
SetupTest LearningModelBindingAPITestGpuSetup;
VoidTest CpuSqueezeNet;
VoidTest CpuSqueezeNetEmptyOutputs;
VoidTest CpuSqueezeNetUnboundOutputs;
VoidTest CpuSqueezeNetBindInputTensorAsInspectable;
VoidTest CastMapInt64;
VoidTest DictionaryVectorizerMapInt64;
VoidTest DictionaryVectorizerMapString;
VoidTest ZipMapInt64;
VoidTest ZipMapInt64Unbound;
VoidTest ZipMapString;
VoidTest GpuSqueezeNet;
VoidTest GpuSqueezeNetEmptyOutputs;
VoidTest GpuSqueezeNetUnboundOutputs;
VoidTest ImageBindingDimensions;
VoidTest VerifyInvalidBindExceptions;
VoidTest BindInvalidInputName;
VoidTest VerifyOutputAfterEvaluateAsyncCalledTwice;
VoidTest VerifyOutputAfterImageBindCalledTwice;
VoidTest SequenceLengthTensorFloat;
VoidTest SequenceConstructTensorString;
};
const LearningModelBindingAPITestApi& getapi();
WINML_TEST_CLASS_BEGIN_WITH_SETUP(LearningModelBindingAPITest, LearningModelBindingAPITestSetup)
WINML_TEST(LearningModelBindingAPITest, CpuSqueezeNet)
WINML_TEST(LearningModelBindingAPITest, CpuSqueezeNetEmptyOutputs)
WINML_TEST(LearningModelBindingAPITest, CpuSqueezeNetUnboundOutputs)
WINML_TEST(LearningModelBindingAPITest, CpuSqueezeNetBindInputTensorAsInspectable)
WINML_TEST(LearningModelBindingAPITest, CastMapInt64)
WINML_TEST(LearningModelBindingAPITest, DictionaryVectorizerMapInt64)
WINML_TEST(LearningModelBindingAPITest, DictionaryVectorizerMapString)
WINML_TEST(LearningModelBindingAPITest, ZipMapInt64)
WINML_TEST(LearningModelBindingAPITest, ZipMapInt64Unbound)
WINML_TEST(LearningModelBindingAPITest, ZipMapString)
WINML_TEST(LearningModelBindingAPITest, VerifyOutputAfterEvaluateAsyncCalledTwice)
WINML_TEST(LearningModelBindingAPITest, VerifyOutputAfterImageBindCalledTwice)
WINML_TEST(LearningModelBindingAPITest, SequenceLengthTensorFloat)
WINML_TEST(LearningModelBindingAPITest, SequenceConstructTensorString)
WINML_TEST_CLASS_END()
WINML_TEST_CLASS_BEGIN_WITH_SETUP(LearningModelBindingAPITestGpu, LearningModelBindingAPITestGpuSetup)
WINML_TEST(LearningModelBindingAPITestGpu, GpuSqueezeNet)
WINML_TEST(LearningModelBindingAPITestGpu, GpuSqueezeNetEmptyOutputs)
WINML_TEST(LearningModelBindingAPITestGpu, GpuSqueezeNetUnboundOutputs)
WINML_TEST(LearningModelBindingAPITestGpu, ImageBindingDimensions)
WINML_TEST(LearningModelBindingAPITestGpu, VerifyInvalidBindExceptions)
WINML_TEST(LearningModelBindingAPITestGpu, BindInvalidInputName)
WINML_TEST_CLASS_END()