mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-26 22:35:43 +00:00
3bf614fd47
* Alloc begin FeedsFetches refactoring Refactor Tensor class Fix buffer deletor Remove new/delete deleted Adjust alloc move Fix up xnnpack provider Clarifying the comment on Create()
70 lines
No EOL
2 KiB
C++
70 lines
No EOL
2 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#pragma once
|
|
|
|
#include <gsl/gsl>
|
|
|
|
namespace onnxruntime {
|
|
// Inspired by Fekir's Blog https://fekir.info/post/span-the-missing-constructor/
|
|
// Used under MIT license
|
|
|
|
// Use AsSpan for less typing on any container including initializer list to create a span
|
|
// (unnamed, untyped initializer list does not automatically convert to gsl::span).
|
|
// {1, 2, 3} as such does not have a type
|
|
// (see https://scottmeyers.blogspot.com/2014/03/if-braced-initializers-have-no-type-why.html)
|
|
//
|
|
// Example: AsSpan({1, 2, 3}) results in gsl::span<const int>
|
|
//
|
|
// The above would deduce to std::initializer_list<int> and the result is gsl::span<const int>
|
|
//
|
|
// AsSpan<int64_t>({1, 2, 3}) produces gsl::span<const int64_t>
|
|
//
|
|
// We can also do std::array<int64_t, 3>{1, 2, 3} that can be automatically converted to span
|
|
// without memory allocation.
|
|
//
|
|
// If type conversion is not required, then for C++17 std::array template parameters are
|
|
// auto-deduced. Example: std::array{1, 2, 3}.
|
|
// We are aiming at not allocating memory dynamically.
|
|
|
|
namespace details {
|
|
template <class P>
|
|
constexpr auto AsSpanImpl(P* p, size_t s) {
|
|
return gsl::span<P>(p, s);
|
|
}
|
|
} // namespace details
|
|
|
|
template <class C>
|
|
constexpr auto AsSpan(C& c) {
|
|
return details::AsSpanImpl(c.data(), c.size());
|
|
}
|
|
|
|
template <class C>
|
|
constexpr auto AsSpan(const C& c) {
|
|
return details::AsSpanImpl(c.data(), c.size());
|
|
}
|
|
|
|
template <class C>
|
|
constexpr auto AsSpan(C&& c) {
|
|
return details::AsSpanImpl(c.data(), c.size());
|
|
}
|
|
|
|
template <class T>
|
|
constexpr auto AsSpan(std::initializer_list<T> c) {
|
|
return details::AsSpanImpl(c.begin(), c.size());
|
|
}
|
|
|
|
template <class T, size_t N>
|
|
constexpr auto AsSpan(T (&arr)[N]) {
|
|
return details::AsSpanImpl(arr, N);
|
|
}
|
|
|
|
template <class T, size_t N>
|
|
constexpr auto AsSpan(const T (&arr)[N]) {
|
|
return details::AsSpanImpl(arr, N);
|
|
}
|
|
|
|
template<class T>
|
|
inline gsl::span<const T> EmptySpan() { return gsl::span<const T>(); }
|
|
|
|
} |