onnxruntime/winml/lib/Api/MapFeatureDescriptor.h
Sheil Kumar 87cb6fd495
Add LearningModelBuilder to WinML Experimental Namespace along with various Audio operators (#6623)
* model building

* fix build

* winml adapter model building api

* model building

* make build

* make build again

* add model building with audio op

* inplace and inorder fft

* add ifft

* works!

* cleanup

* add comments

* switch to iterative rather than recursive and use parallelization

* batched parallelization

* fft->dft

* cleanup

* window functions

* add melweightmatrix op

* updates to make spectrogram test work

* push latest

* add onesided

* cleanup

* Clean up building apis and fix mel

* cleanup

* cleanup

* naive stft

* fix test output

* middle c complete

* 3 tones

* cleanup

* signal def new line

* Add save functionality

* Perf improvements, 10x improvement

* cleanup

* use bitreverse lookup table for performance

* implement constant initializers for tensors

* small changes

* add matmul tests

* merge issues

* support add attribute

* add tests for double data type windowfunctions and minor cleanup

* stft onesided/and not tests

* cleanup

* cleanup

* clean up

* cleanup

* remove threading attribute

* forward declare orttypeinfo

* warnings

* fwd declare

* fix warnings

* 1 more warning

* remove saving to e drive...

* cleanup and fix stft test

* add opset picker

* small additions

* add onnxruntime tests

* add signed/unsigned

* fix warning

* fix warning

* finish onnxruntime tests

* make windows namespace build succeed

* add experimental flag

* add experimental api into nuget package

* add experimental api build flag and add to windows ai nuget package

* turn experimental for tests

* add minimum opset version to new experimental domain

* api cleanup

* disable ms experimental ops test when --ms_experimental is not enabled

* add macro behind flag

* remove unused x

* pr feedback

Co-authored-by: Sheil Kumar <sheilk@microsoft.com>
2021-02-12 14:17:10 -08:00

65 lines
No EOL
1.4 KiB
C++

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#pragma once
#include "MapFeatureDescriptor.g.h"
#include "iengine.h"
namespace WINMLP {
struct MapFeatureDescriptor : MapFeatureDescriptorT<
MapFeatureDescriptor,
ILearningModelFeatureDescriptorNative,
_winml::IDescriptorInfoProvider> {
MapFeatureDescriptor() = delete;
MapFeatureDescriptor(
const char* name,
const char* description,
bool is_required,
winml::TensorKind keyKind,
winml::ILearningModelFeatureDescriptor valueKind);
// IMapDescriptor
winml::TensorKind
KeyKind();
winml::ILearningModelFeatureDescriptor
ValueDescriptor();
// IFeatureDescriptor
hstring
Name();
hstring
Description();
winml::LearningModelFeatureKind
Kind();
bool
IsRequired();
STDMETHOD(GetName)
(
const wchar_t** name,
uint32_t* cchName) override;
STDMETHOD(GetDescription)
(
const wchar_t** description,
uint32_t* cchDescription) override;
STDMETHOD(GetDescriptorInfo)
(
_winml::IEngineFactory* engine_factory,
_winml::IDescriptorInfo** info) override;
private:
winrt::hstring name_;
winrt::hstring description_;
bool is_required_;
winml::TensorKind key_kind_;
winml::ILearningModelFeatureDescriptor value_kind_;
};
} // namespace WINMLP