mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-27 22:45:57 +00:00
0f5f17c175
* Pass flags to build/test WinML in CI * Add initial CMake config for unit tests in WinML * Set winml_unittests standard to C++17 * Add WinML API tests and port them to googletest * Install WinML test collateral * Add LearningModelSessionAPITests ported to googletest * Fix WinML test files encoding * Add GPU tests * Add parameterized test, skip GPU tests * Enable precompiled header * Remove unused code and collateral * Remove brand images * Add dllload.cpp * Remove images not used in API tests * Add LICENSE.md to image collaterals * Add models with licenses * Remove FNS Candy tests * Add API test models * Add ModelInSubdirectory * Install collaterals post-build with copy_if_different, split common lib * fix warnings * Link to gtest_main
23 lines
989 B
C++
23 lines
989 B
C++
//-----------------------------------------------------------------------------
|
|
//
|
|
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
|
|
#pragma once
|
|
|
|
#include "std.h"
|
|
|
|
namespace ProtobufHelpers
|
|
{
|
|
// LoadTensorFromProtobufFile take a path to a FP32 data file and loads it into a 32bit array or
|
|
// 16bit array based on isFp16
|
|
winrt::Windows::AI::MachineLearning::ITensor LoadTensorFromProtobufFile(const std::wstring& filePath, bool isFp16);
|
|
// LoadTensorFloat16FromProtobufFile takes a path to a FP16 data file and loads it into a 16bit array
|
|
winrt::Windows::AI::MachineLearning::TensorFloat16Bit LoadTensorFloat16FromProtobufFile(const std::wstring& filePath);
|
|
|
|
winrt::Windows::AI::MachineLearning::LearningModel CreateModel(
|
|
winrt::Windows::AI::MachineLearning::TensorKind kind,
|
|
const std::vector<int64_t>& shape,
|
|
uint32_t num_elements = 1);
|
|
}
|