mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
87cb6fd495
* 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>
92 lines
No EOL
2.9 KiB
C++
92 lines
No EOL
2.9 KiB
C++
#include "pch.h"
|
|
#include "LearningModelOperator.h"
|
|
|
|
namespace WINML_EXPERIMENTALP {
|
|
|
|
static uint32_t c_operator_index = 0;
|
|
|
|
LearningModelOperator::LearningModelOperator(hstring const& type) :
|
|
LearningModelOperator(type, L"")
|
|
{}
|
|
|
|
LearningModelOperator::LearningModelOperator(hstring const& type, hstring const& domain) :
|
|
type_(type),
|
|
domain_(domain) {
|
|
constant_input_mapping_ = winrt::single_threaded_map<winrt::hstring, wf::IInspectable>();
|
|
input_mapping_ = winrt::single_threaded_map<winrt::hstring, winrt::hstring>();
|
|
output_mapping_ = winrt::single_threaded_map<winrt::hstring, winrt::hstring>();
|
|
attribute_values_ = winrt::single_threaded_map<winrt::hstring, wf::IInspectable>();
|
|
|
|
SetName(L"");
|
|
}
|
|
|
|
winml_experimental::LearningModelOperator LearningModelOperator::SetName(hstring const& name) {
|
|
if (name.empty()) {
|
|
std::wostringstream name_stream;
|
|
name_stream << type_.c_str() << "_" << c_operator_index++;
|
|
name_ = name_stream.str().c_str();
|
|
} else {
|
|
name_ = name;
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
winml_experimental::LearningModelOperator LearningModelOperator::SetInput(
|
|
hstring const& operator_input_name, hstring const& input_name) {
|
|
|
|
// TODO Validate against allowed operator input NAMES. The types are not deduced.
|
|
input_mapping_.Insert(operator_input_name, input_name);
|
|
return *this;
|
|
}
|
|
|
|
winml_experimental::LearningModelOperator LearningModelOperator::SetConstant(
|
|
hstring const& operator_input_name, wf::IInspectable const& value) {
|
|
// TODO Validate against allowed operator input NAMES. The types are not deduced.
|
|
auto constant_name = name_ + L"." + operator_input_name;
|
|
input_mapping_.Insert(operator_input_name, constant_name);
|
|
constant_input_mapping_.Insert(constant_name, value);
|
|
return *this;
|
|
}
|
|
|
|
winml_experimental::LearningModelOperator LearningModelOperator::SetOutput(
|
|
hstring const& operator_output_name, hstring const& output_name) {
|
|
// TODO Validate against allowed operator output NAMES. The types are not deduced.
|
|
output_mapping_.Insert(operator_output_name, output_name);
|
|
return *this;
|
|
}
|
|
|
|
winml_experimental::LearningModelOperator LearningModelOperator::SetAttribute(
|
|
hstring const& name, Windows::Foundation::IInspectable const& value) {
|
|
attribute_values_.Insert(name, value);
|
|
return *this;
|
|
}
|
|
|
|
hstring LearningModelOperator::Name() {
|
|
return name_;
|
|
}
|
|
|
|
hstring LearningModelOperator::Type() {
|
|
return type_;
|
|
}
|
|
|
|
hstring LearningModelOperator::Domain() {
|
|
return domain_;
|
|
}
|
|
|
|
wfc::IMap<winrt::hstring, winrt::hstring> LearningModelOperator::InputMapping(){
|
|
return input_mapping_;
|
|
}
|
|
|
|
wfc::IMap<winrt::hstring, wf::IInspectable> LearningModelOperator::ConstantInputMapping() {
|
|
return constant_input_mapping_;
|
|
}
|
|
|
|
wfc::IMap<winrt::hstring, winrt::hstring> LearningModelOperator::OutputMapping() {
|
|
return output_mapping_;
|
|
}
|
|
|
|
wfc::IMap<winrt::hstring, wf::IInspectable> LearningModelOperator::AttributeMap() {
|
|
return attribute_values_;
|
|
}
|
|
|
|
} // namespace WINML_EXPERIMENTALP
|