449 lines
43 KiB
HTML
449 lines
43 KiB
HTML
<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN">
|
|
|
|
<html lang="en">
|
|
|
|
<head>
|
|
<meta http-equiv="Content-Type" content="text/html; charset=UTF-8">
|
|
<title>LCOV - coverage.info - libtorch/include/c10/util/ArrayRef.h</title>
|
|
<link rel="stylesheet" type="text/css" href="../../../../gcov.css">
|
|
</head>
|
|
|
|
<body>
|
|
|
|
<table width="100%" border=0 cellspacing=0 cellpadding=0>
|
|
<tr><td class="title">LCOV - code coverage report</td></tr>
|
|
<tr><td class="ruler"><img src="../../../../glass.png" width=3 height=3 alt=""></td></tr>
|
|
|
|
<tr>
|
|
<td width="100%">
|
|
<table cellpadding=1 border=0 width="100%">
|
|
<tr>
|
|
<td width="10%" class="headerItem">Current view:</td>
|
|
<td width="10%" class="headerValue"><a href="../../../../index.html">top level</a> - <a href="index.html">libtorch/include/c10/util</a> - ArrayRef.h<span style="font-size: 80%;"> (source / <a href="ArrayRef.h.func-c.html">functions</a>)</span></td>
|
|
<td width="5%"></td>
|
|
<td width="5%"></td>
|
|
<td width="5%" class="headerCovTableHead">Coverage</td>
|
|
<td width="5%" class="headerCovTableHead" title="Covered + Uncovered code">Total</td>
|
|
<td width="5%" class="headerCovTableHead" title="Exercised code only">Hit</td>
|
|
</tr>
|
|
<tr>
|
|
<td class="headerItem">Test:</td>
|
|
<td class="headerValue">coverage.info</td>
|
|
<td></td>
|
|
<td class="headerItem">Lines:</td>
|
|
<td class="headerCovTableEntryHi">100.0 %</td>
|
|
<td class="headerCovTableEntry">35</td>
|
|
<td class="headerCovTableEntry">35</td>
|
|
</tr>
|
|
<tr>
|
|
<td class="headerItem">Test Date:</td>
|
|
<td class="headerValue">2024-04-30 13:17:26</td>
|
|
<td></td>
|
|
<td class="headerItem">Functions:</td>
|
|
<td class="headerCovTableEntryHi">100.0 %</td>
|
|
<td class="headerCovTableEntry">21</td>
|
|
<td class="headerCovTableEntry">21</td>
|
|
</tr>
|
|
<tr><td><img src="../../../../glass.png" width=3 height=3 alt=""></td></tr>
|
|
</table>
|
|
</td>
|
|
</tr>
|
|
|
|
<tr><td class="ruler"><img src="../../../../glass.png" width=3 height=3 alt=""></td></tr>
|
|
</table>
|
|
|
|
<table cellpadding=0 cellspacing=0 border=0>
|
|
<tr>
|
|
<td><br></td>
|
|
</tr>
|
|
<tr>
|
|
<td>
|
|
<pre class="sourceHeading"> Line data Source code</pre>
|
|
<pre class="source">
|
|
<span id="L1"><span class="lineNum"> 1</span> : //===--- ArrayRef.h - Array Reference Wrapper -------------------*- C++ -*-===//</span>
|
|
<span id="L2"><span class="lineNum"> 2</span> : //</span>
|
|
<span id="L3"><span class="lineNum"> 3</span> : // The LLVM Compiler Infrastructure</span>
|
|
<span id="L4"><span class="lineNum"> 4</span> : //</span>
|
|
<span id="L5"><span class="lineNum"> 5</span> : // This file is distributed under the University of Illinois Open Source</span>
|
|
<span id="L6"><span class="lineNum"> 6</span> : // License. See LICENSE.TXT for details.</span>
|
|
<span id="L7"><span class="lineNum"> 7</span> : //</span>
|
|
<span id="L8"><span class="lineNum"> 8</span> : //===----------------------------------------------------------------------===//</span>
|
|
<span id="L9"><span class="lineNum"> 9</span> : </span>
|
|
<span id="L10"><span class="lineNum"> 10</span> : // ATen: modified from llvm::ArrayRef.</span>
|
|
<span id="L11"><span class="lineNum"> 11</span> : // removed llvm-specific functionality</span>
|
|
<span id="L12"><span class="lineNum"> 12</span> : // removed some implicit const -> non-const conversions that rely on</span>
|
|
<span id="L13"><span class="lineNum"> 13</span> : // complicated std::enable_if meta-programming</span>
|
|
<span id="L14"><span class="lineNum"> 14</span> : // removed a bunch of slice variants for simplicity...</span>
|
|
<span id="L15"><span class="lineNum"> 15</span> : </span>
|
|
<span id="L16"><span class="lineNum"> 16</span> : #pragma once</span>
|
|
<span id="L17"><span class="lineNum"> 17</span> : </span>
|
|
<span id="L18"><span class="lineNum"> 18</span> : #include <c10/util/C++17.h></span>
|
|
<span id="L19"><span class="lineNum"> 19</span> : #include <c10/util/Deprecated.h></span>
|
|
<span id="L20"><span class="lineNum"> 20</span> : #include <c10/util/Exception.h></span>
|
|
<span id="L21"><span class="lineNum"> 21</span> : #include <c10/util/SmallVector.h></span>
|
|
<span id="L22"><span class="lineNum"> 22</span> : </span>
|
|
<span id="L23"><span class="lineNum"> 23</span> : #include <array></span>
|
|
<span id="L24"><span class="lineNum"> 24</span> : #include <iterator></span>
|
|
<span id="L25"><span class="lineNum"> 25</span> : #include <vector></span>
|
|
<span id="L26"><span class="lineNum"> 26</span> : </span>
|
|
<span id="L27"><span class="lineNum"> 27</span> : namespace c10 {</span>
|
|
<span id="L28"><span class="lineNum"> 28</span> : /// ArrayRef - Represent a constant reference to an array (0 or more elements</span>
|
|
<span id="L29"><span class="lineNum"> 29</span> : /// consecutively in memory), i.e. a start pointer and a length. It allows</span>
|
|
<span id="L30"><span class="lineNum"> 30</span> : /// various APIs to take consecutive elements easily and conveniently.</span>
|
|
<span id="L31"><span class="lineNum"> 31</span> : ///</span>
|
|
<span id="L32"><span class="lineNum"> 32</span> : /// This class does not own the underlying data, it is expected to be used in</span>
|
|
<span id="L33"><span class="lineNum"> 33</span> : /// situations where the data resides in some other buffer, whose lifetime</span>
|
|
<span id="L34"><span class="lineNum"> 34</span> : /// extends past that of the ArrayRef. For this reason, it is not in general</span>
|
|
<span id="L35"><span class="lineNum"> 35</span> : /// safe to store an ArrayRef.</span>
|
|
<span id="L36"><span class="lineNum"> 36</span> : ///</span>
|
|
<span id="L37"><span class="lineNum"> 37</span> : /// This is intended to be trivially copyable, so it should be passed by</span>
|
|
<span id="L38"><span class="lineNum"> 38</span> : /// value.</span>
|
|
<span id="L39"><span class="lineNum"> 39</span> : template <typename T></span>
|
|
<span id="L40"><span class="lineNum"> 40</span> : class ArrayRef final {</span>
|
|
<span id="L41"><span class="lineNum"> 41</span> : public:</span>
|
|
<span id="L42"><span class="lineNum"> 42</span> : using iterator = const T*;</span>
|
|
<span id="L43"><span class="lineNum"> 43</span> : using const_iterator = const T*;</span>
|
|
<span id="L44"><span class="lineNum"> 44</span> : using size_type = size_t;</span>
|
|
<span id="L45"><span class="lineNum"> 45</span> : using value_type = T;</span>
|
|
<span id="L46"><span class="lineNum"> 46</span> : </span>
|
|
<span id="L47"><span class="lineNum"> 47</span> : using reverse_iterator = std::reverse_iterator<iterator>;</span>
|
|
<span id="L48"><span class="lineNum"> 48</span> : </span>
|
|
<span id="L49"><span class="lineNum"> 49</span> : private:</span>
|
|
<span id="L50"><span class="lineNum"> 50</span> : /// The start of the array, in an external buffer.</span>
|
|
<span id="L51"><span class="lineNum"> 51</span> : const T* Data;</span>
|
|
<span id="L52"><span class="lineNum"> 52</span> : </span>
|
|
<span id="L53"><span class="lineNum"> 53</span> : /// The number of elements.</span>
|
|
<span id="L54"><span class="lineNum"> 54</span> : size_type Length;</span>
|
|
<span id="L55"><span class="lineNum"> 55</span> : </span>
|
|
<span id="L56"><span class="lineNum"> 56</span> <span class="tlaGNC tlaBgGNC"> 106536 : void debugCheckNullptrInvariant() {</span></span>
|
|
<span id="L57"><span class="lineNum"> 57</span> <span class="tlaGNC"> 106536 : TORCH_INTERNAL_ASSERT_DEBUG_ONLY(</span></span>
|
|
<span id="L58"><span class="lineNum"> 58</span> : Data != nullptr || Length == 0,</span>
|
|
<span id="L59"><span class="lineNum"> 59</span> : "created ArrayRef with nullptr and non-zero length! c10::optional relies on this being illegal");</span>
|
|
<span id="L60"><span class="lineNum"> 60</span> <span class="tlaGNC"> 106536 : }</span></span>
|
|
<span id="L61"><span class="lineNum"> 61</span> : </span>
|
|
<span id="L62"><span class="lineNum"> 62</span> : public:</span>
|
|
<span id="L63"><span class="lineNum"> 63</span> : /// @name Constructors</span>
|
|
<span id="L64"><span class="lineNum"> 64</span> : /// @{</span>
|
|
<span id="L65"><span class="lineNum"> 65</span> : </span>
|
|
<span id="L66"><span class="lineNum"> 66</span> : /// Construct an empty ArrayRef.</span>
|
|
<span id="L67"><span class="lineNum"> 67</span> : /* implicit */ constexpr ArrayRef() : Data(nullptr), Length(0) {}</span>
|
|
<span id="L68"><span class="lineNum"> 68</span> : </span>
|
|
<span id="L69"><span class="lineNum"> 69</span> : /// Construct an ArrayRef from a single element.</span>
|
|
<span id="L70"><span class="lineNum"> 70</span> : // TODO Make this explicit</span>
|
|
<span id="L71"><span class="lineNum"> 71</span> <span class="tlaGNC"> 90643162 : constexpr ArrayRef(const T& OneElt) : Data(&OneElt), Length(1) {}</span></span>
|
|
<span id="L72"><span class="lineNum"> 72</span> : </span>
|
|
<span id="L73"><span class="lineNum"> 73</span> : /// Construct an ArrayRef from a pointer and length.</span>
|
|
<span id="L74"><span class="lineNum"> 74</span> <span class="tlaGNC"> 106536 : C10_HOST_CONSTEXPR_EXCEPT_WIN_CUDA ArrayRef(const T* data, size_t length)</span></span>
|
|
<span id="L75"><span class="lineNum"> 75</span> <span class="tlaGNC"> 106536 : : Data(data), Length(length) {</span></span>
|
|
<span id="L76"><span class="lineNum"> 76</span> <span class="tlaGNC"> 106536 : debugCheckNullptrInvariant();</span></span>
|
|
<span id="L77"><span class="lineNum"> 77</span> <span class="tlaGNC"> 106536 : }</span></span>
|
|
<span id="L78"><span class="lineNum"> 78</span> : </span>
|
|
<span id="L79"><span class="lineNum"> 79</span> : /// Construct an ArrayRef from a range.</span>
|
|
<span id="L80"><span class="lineNum"> 80</span> : C10_HOST_CONSTEXPR_EXCEPT_WIN_CUDA ArrayRef(const T* begin, const T* end)</span>
|
|
<span id="L81"><span class="lineNum"> 81</span> : : Data(begin), Length(end - begin) {</span>
|
|
<span id="L82"><span class="lineNum"> 82</span> : debugCheckNullptrInvariant();</span>
|
|
<span id="L83"><span class="lineNum"> 83</span> : }</span>
|
|
<span id="L84"><span class="lineNum"> 84</span> : </span>
|
|
<span id="L85"><span class="lineNum"> 85</span> : /// Construct an ArrayRef from a SmallVector. This is templated in order to</span>
|
|
<span id="L86"><span class="lineNum"> 86</span> : /// avoid instantiating SmallVectorTemplateCommon<T> whenever we</span>
|
|
<span id="L87"><span class="lineNum"> 87</span> : /// copy-construct an ArrayRef.</span>
|
|
<span id="L88"><span class="lineNum"> 88</span> : template <typename U></span>
|
|
<span id="L89"><span class="lineNum"> 89</span> : /* implicit */ ArrayRef(const SmallVectorTemplateCommon<T, U>& Vec)</span>
|
|
<span id="L90"><span class="lineNum"> 90</span> : : Data(Vec.data()), Length(Vec.size()) {</span>
|
|
<span id="L91"><span class="lineNum"> 91</span> : debugCheckNullptrInvariant();</span>
|
|
<span id="L92"><span class="lineNum"> 92</span> : }</span>
|
|
<span id="L93"><span class="lineNum"> 93</span> : </span>
|
|
<span id="L94"><span class="lineNum"> 94</span> : template <</span>
|
|
<span id="L95"><span class="lineNum"> 95</span> : typename Container,</span>
|
|
<span id="L96"><span class="lineNum"> 96</span> : typename = std::enable_if_t<std::is_same<</span>
|
|
<span id="L97"><span class="lineNum"> 97</span> : std::remove_const_t<decltype(std::declval<Container>().data())>,</span>
|
|
<span id="L98"><span class="lineNum"> 98</span> : T*>::value>></span>
|
|
<span id="L99"><span class="lineNum"> 99</span> : /* implicit */ ArrayRef(const Container& container)</span>
|
|
<span id="L100"><span class="lineNum"> 100</span> : : Data(container.data()), Length(container.size()) {</span>
|
|
<span id="L101"><span class="lineNum"> 101</span> : debugCheckNullptrInvariant();</span>
|
|
<span id="L102"><span class="lineNum"> 102</span> : }</span>
|
|
<span id="L103"><span class="lineNum"> 103</span> : </span>
|
|
<span id="L104"><span class="lineNum"> 104</span> : /// Construct an ArrayRef from a std::vector.</span>
|
|
<span id="L105"><span class="lineNum"> 105</span> : // The enable_if stuff here makes sure that this isn't used for</span>
|
|
<span id="L106"><span class="lineNum"> 106</span> : // std::vector<bool>, because ArrayRef can't work on a std::vector<bool></span>
|
|
<span id="L107"><span class="lineNum"> 107</span> : // bitfield.</span>
|
|
<span id="L108"><span class="lineNum"> 108</span> : template <typename A></span>
|
|
<span id="L109"><span class="lineNum"> 109</span> <span class="tlaGNC"> 415512 : /* implicit */ ArrayRef(const std::vector<T, A>& Vec)</span></span>
|
|
<span id="L110"><span class="lineNum"> 110</span> <span class="tlaGNC"> 415512 : : Data(Vec.data()), Length(Vec.size()) {</span></span>
|
|
<span id="L111"><span class="lineNum"> 111</span> : static_assert(</span>
|
|
<span id="L112"><span class="lineNum"> 112</span> : !std::is_same<T, bool>::value,</span>
|
|
<span id="L113"><span class="lineNum"> 113</span> : "ArrayRef<bool> cannot be constructed from a std::vector<bool> bitfield.");</span>
|
|
<span id="L114"><span class="lineNum"> 114</span> <span class="tlaGNC"> 415512 : }</span></span>
|
|
<span id="L115"><span class="lineNum"> 115</span> : </span>
|
|
<span id="L116"><span class="lineNum"> 116</span> : /// Construct an ArrayRef from a std::array</span>
|
|
<span id="L117"><span class="lineNum"> 117</span> : template <size_t N></span>
|
|
<span id="L118"><span class="lineNum"> 118</span> : /* implicit */ constexpr ArrayRef(const std::array<T, N>& Arr)</span>
|
|
<span id="L119"><span class="lineNum"> 119</span> : : Data(Arr.data()), Length(N) {}</span>
|
|
<span id="L120"><span class="lineNum"> 120</span> : </span>
|
|
<span id="L121"><span class="lineNum"> 121</span> : /// Construct an ArrayRef from a C array.</span>
|
|
<span id="L122"><span class="lineNum"> 122</span> : template <size_t N></span>
|
|
<span id="L123"><span class="lineNum"> 123</span> : /* implicit */ constexpr ArrayRef(const T (&Arr)[N]) : Data(Arr), Length(N) {}</span>
|
|
<span id="L124"><span class="lineNum"> 124</span> : </span>
|
|
<span id="L125"><span class="lineNum"> 125</span> : /// Construct an ArrayRef from a std::initializer_list.</span>
|
|
<span id="L126"><span class="lineNum"> 126</span> <span class="tlaGNC"> 3934 : /* implicit */ constexpr ArrayRef(const std::initializer_list<T>& Vec)</span></span>
|
|
<span id="L127"><span class="lineNum"> 127</span> <span class="tlaGNC"> 3934 : : Data(</span></span>
|
|
<span id="L128"><span class="lineNum"> 128</span> <span class="tlaGNC"> 3934 : std::begin(Vec) == std::end(Vec) ? static_cast<T*>(nullptr)</span></span>
|
|
<span id="L129"><span class="lineNum"> 129</span> <span class="tlaGNC"> 3934 : : std::begin(Vec)),</span></span>
|
|
<span id="L130"><span class="lineNum"> 130</span> <span class="tlaGNC"> 3934 : Length(Vec.size()) {}</span></span>
|
|
<span id="L131"><span class="lineNum"> 131</span> : </span>
|
|
<span id="L132"><span class="lineNum"> 132</span> : /// @}</span>
|
|
<span id="L133"><span class="lineNum"> 133</span> : /// @name Simple Operations</span>
|
|
<span id="L134"><span class="lineNum"> 134</span> : /// @{</span>
|
|
<span id="L135"><span class="lineNum"> 135</span> : </span>
|
|
<span id="L136"><span class="lineNum"> 136</span> <span class="tlaGNC"> 17786 : constexpr iterator begin() const {</span></span>
|
|
<span id="L137"><span class="lineNum"> 137</span> <span class="tlaGNC"> 17786 : return Data;</span></span>
|
|
<span id="L138"><span class="lineNum"> 138</span> : }</span>
|
|
<span id="L139"><span class="lineNum"> 139</span> <span class="tlaGNC"> 17786 : constexpr iterator end() const {</span></span>
|
|
<span id="L140"><span class="lineNum"> 140</span> <span class="tlaGNC"> 17786 : return Data + Length;</span></span>
|
|
<span id="L141"><span class="lineNum"> 141</span> : }</span>
|
|
<span id="L142"><span class="lineNum"> 142</span> : </span>
|
|
<span id="L143"><span class="lineNum"> 143</span> : // These are actually the same as iterator, since ArrayRef only</span>
|
|
<span id="L144"><span class="lineNum"> 144</span> : // gives you const iterators.</span>
|
|
<span id="L145"><span class="lineNum"> 145</span> : constexpr const_iterator cbegin() const {</span>
|
|
<span id="L146"><span class="lineNum"> 146</span> : return Data;</span>
|
|
<span id="L147"><span class="lineNum"> 147</span> : }</span>
|
|
<span id="L148"><span class="lineNum"> 148</span> : constexpr const_iterator cend() const {</span>
|
|
<span id="L149"><span class="lineNum"> 149</span> : return Data + Length;</span>
|
|
<span id="L150"><span class="lineNum"> 150</span> : }</span>
|
|
<span id="L151"><span class="lineNum"> 151</span> : </span>
|
|
<span id="L152"><span class="lineNum"> 152</span> : constexpr reverse_iterator rbegin() const {</span>
|
|
<span id="L153"><span class="lineNum"> 153</span> : return reverse_iterator(end());</span>
|
|
<span id="L154"><span class="lineNum"> 154</span> : }</span>
|
|
<span id="L155"><span class="lineNum"> 155</span> : constexpr reverse_iterator rend() const {</span>
|
|
<span id="L156"><span class="lineNum"> 156</span> : return reverse_iterator(begin());</span>
|
|
<span id="L157"><span class="lineNum"> 157</span> : }</span>
|
|
<span id="L158"><span class="lineNum"> 158</span> : </span>
|
|
<span id="L159"><span class="lineNum"> 159</span> : /// empty - Check if the array is empty.</span>
|
|
<span id="L160"><span class="lineNum"> 160</span> : constexpr bool empty() const {</span>
|
|
<span id="L161"><span class="lineNum"> 161</span> : return Length == 0;</span>
|
|
<span id="L162"><span class="lineNum"> 162</span> : }</span>
|
|
<span id="L163"><span class="lineNum"> 163</span> : </span>
|
|
<span id="L164"><span class="lineNum"> 164</span> <span class="tlaGNC"> 17910 : constexpr const T* data() const {</span></span>
|
|
<span id="L165"><span class="lineNum"> 165</span> <span class="tlaGNC"> 17910 : return Data;</span></span>
|
|
<span id="L166"><span class="lineNum"> 166</span> : }</span>
|
|
<span id="L167"><span class="lineNum"> 167</span> : </span>
|
|
<span id="L168"><span class="lineNum"> 168</span> : /// size - Get the array size.</span>
|
|
<span id="L169"><span class="lineNum"> 169</span> <span class="tlaGNC"> 419068 : constexpr size_t size() const {</span></span>
|
|
<span id="L170"><span class="lineNum"> 170</span> <span class="tlaGNC"> 419068 : return Length;</span></span>
|
|
<span id="L171"><span class="lineNum"> 171</span> : }</span>
|
|
<span id="L172"><span class="lineNum"> 172</span> : </span>
|
|
<span id="L173"><span class="lineNum"> 173</span> : /// front - Get the first element.</span>
|
|
<span id="L174"><span class="lineNum"> 174</span> : C10_HOST_CONSTEXPR_EXCEPT_WIN_CUDA const T& front() const {</span>
|
|
<span id="L175"><span class="lineNum"> 175</span> : TORCH_CHECK(</span>
|
|
<span id="L176"><span class="lineNum"> 176</span> : !empty(), "ArrayRef: attempted to access front() of empty list");</span>
|
|
<span id="L177"><span class="lineNum"> 177</span> : return Data[0];</span>
|
|
<span id="L178"><span class="lineNum"> 178</span> : }</span>
|
|
<span id="L179"><span class="lineNum"> 179</span> : </span>
|
|
<span id="L180"><span class="lineNum"> 180</span> : /// back - Get the last element.</span>
|
|
<span id="L181"><span class="lineNum"> 181</span> : C10_HOST_CONSTEXPR_EXCEPT_WIN_CUDA const T& back() const {</span>
|
|
<span id="L182"><span class="lineNum"> 182</span> : TORCH_CHECK(!empty(), "ArrayRef: attempted to access back() of empty list");</span>
|
|
<span id="L183"><span class="lineNum"> 183</span> : return Data[Length - 1];</span>
|
|
<span id="L184"><span class="lineNum"> 184</span> : }</span>
|
|
<span id="L185"><span class="lineNum"> 185</span> : </span>
|
|
<span id="L186"><span class="lineNum"> 186</span> : /// equals - Check for element-wise equality.</span>
|
|
<span id="L187"><span class="lineNum"> 187</span> : constexpr bool equals(ArrayRef RHS) const {</span>
|
|
<span id="L188"><span class="lineNum"> 188</span> : return Length == RHS.Length && std::equal(begin(), end(), RHS.begin());</span>
|
|
<span id="L189"><span class="lineNum"> 189</span> : }</span>
|
|
<span id="L190"><span class="lineNum"> 190</span> : </span>
|
|
<span id="L191"><span class="lineNum"> 191</span> : /// slice(n, m) - Take M elements of the array starting at element N</span>
|
|
<span id="L192"><span class="lineNum"> 192</span> : C10_HOST_CONSTEXPR_EXCEPT_WIN_CUDA ArrayRef<T> slice(size_t N, size_t M)</span>
|
|
<span id="L193"><span class="lineNum"> 193</span> : const {</span>
|
|
<span id="L194"><span class="lineNum"> 194</span> : TORCH_CHECK(</span>
|
|
<span id="L195"><span class="lineNum"> 195</span> : N + M <= size(),</span>
|
|
<span id="L196"><span class="lineNum"> 196</span> : "ArrayRef: invalid slice, N = ",</span>
|
|
<span id="L197"><span class="lineNum"> 197</span> : N,</span>
|
|
<span id="L198"><span class="lineNum"> 198</span> : "; M = ",</span>
|
|
<span id="L199"><span class="lineNum"> 199</span> : M,</span>
|
|
<span id="L200"><span class="lineNum"> 200</span> : "; size = ",</span>
|
|
<span id="L201"><span class="lineNum"> 201</span> : size());</span>
|
|
<span id="L202"><span class="lineNum"> 202</span> : return ArrayRef<T>(data() + N, M);</span>
|
|
<span id="L203"><span class="lineNum"> 203</span> : }</span>
|
|
<span id="L204"><span class="lineNum"> 204</span> : </span>
|
|
<span id="L205"><span class="lineNum"> 205</span> : /// slice(n) - Chop off the first N elements of the array.</span>
|
|
<span id="L206"><span class="lineNum"> 206</span> : C10_HOST_CONSTEXPR_EXCEPT_WIN_CUDA ArrayRef<T> slice(size_t N) const {</span>
|
|
<span id="L207"><span class="lineNum"> 207</span> : TORCH_CHECK(</span>
|
|
<span id="L208"><span class="lineNum"> 208</span> : N <= size(), "ArrayRef: invalid slice, N = ", N, "; size = ", size());</span>
|
|
<span id="L209"><span class="lineNum"> 209</span> : return slice(N, size() - N);</span>
|
|
<span id="L210"><span class="lineNum"> 210</span> : }</span>
|
|
<span id="L211"><span class="lineNum"> 211</span> : </span>
|
|
<span id="L212"><span class="lineNum"> 212</span> : /// @}</span>
|
|
<span id="L213"><span class="lineNum"> 213</span> : /// @name Operator Overloads</span>
|
|
<span id="L214"><span class="lineNum"> 214</span> : /// @{</span>
|
|
<span id="L215"><span class="lineNum"> 215</span> <span class="tlaGNC"> 88590 : constexpr const T& operator[](size_t Index) const {</span></span>
|
|
<span id="L216"><span class="lineNum"> 216</span> <span class="tlaGNC"> 88590 : return Data[Index];</span></span>
|
|
<span id="L217"><span class="lineNum"> 217</span> : }</span>
|
|
<span id="L218"><span class="lineNum"> 218</span> : </span>
|
|
<span id="L219"><span class="lineNum"> 219</span> : /// Vector compatibility</span>
|
|
<span id="L220"><span class="lineNum"> 220</span> : C10_HOST_CONSTEXPR_EXCEPT_WIN_CUDA const T& at(size_t Index) const {</span>
|
|
<span id="L221"><span class="lineNum"> 221</span> : TORCH_CHECK(</span>
|
|
<span id="L222"><span class="lineNum"> 222</span> : Index < Length,</span>
|
|
<span id="L223"><span class="lineNum"> 223</span> : "ArrayRef: invalid index Index = ",</span>
|
|
<span id="L224"><span class="lineNum"> 224</span> : Index,</span>
|
|
<span id="L225"><span class="lineNum"> 225</span> : "; Length = ",</span>
|
|
<span id="L226"><span class="lineNum"> 226</span> : Length);</span>
|
|
<span id="L227"><span class="lineNum"> 227</span> : return Data[Index];</span>
|
|
<span id="L228"><span class="lineNum"> 228</span> : }</span>
|
|
<span id="L229"><span class="lineNum"> 229</span> : </span>
|
|
<span id="L230"><span class="lineNum"> 230</span> : /// Disallow accidental assignment from a temporary.</span>
|
|
<span id="L231"><span class="lineNum"> 231</span> : ///</span>
|
|
<span id="L232"><span class="lineNum"> 232</span> : /// The declaration here is extra complicated so that "arrayRef = {}"</span>
|
|
<span id="L233"><span class="lineNum"> 233</span> : /// continues to select the move assignment operator.</span>
|
|
<span id="L234"><span class="lineNum"> 234</span> : template <typename U></span>
|
|
<span id="L235"><span class="lineNum"> 235</span> : typename std::enable_if<std::is_same<U, T>::value, ArrayRef<T>>::type&</span>
|
|
<span id="L236"><span class="lineNum"> 236</span> : operator=(U&& Temporary) = delete;</span>
|
|
<span id="L237"><span class="lineNum"> 237</span> : </span>
|
|
<span id="L238"><span class="lineNum"> 238</span> : /// Disallow accidental assignment from a temporary.</span>
|
|
<span id="L239"><span class="lineNum"> 239</span> : ///</span>
|
|
<span id="L240"><span class="lineNum"> 240</span> : /// The declaration here is extra complicated so that "arrayRef = {}"</span>
|
|
<span id="L241"><span class="lineNum"> 241</span> : /// continues to select the move assignment operator.</span>
|
|
<span id="L242"><span class="lineNum"> 242</span> : template <typename U></span>
|
|
<span id="L243"><span class="lineNum"> 243</span> : typename std::enable_if<std::is_same<U, T>::value, ArrayRef<T>>::type&</span>
|
|
<span id="L244"><span class="lineNum"> 244</span> : operator=(std::initializer_list<U>) = delete;</span>
|
|
<span id="L245"><span class="lineNum"> 245</span> : </span>
|
|
<span id="L246"><span class="lineNum"> 246</span> : /// @}</span>
|
|
<span id="L247"><span class="lineNum"> 247</span> : /// @name Expensive Operations</span>
|
|
<span id="L248"><span class="lineNum"> 248</span> : /// @{</span>
|
|
<span id="L249"><span class="lineNum"> 249</span> : std::vector<T> vec() const {</span>
|
|
<span id="L250"><span class="lineNum"> 250</span> : return std::vector<T>(Data, Data + Length);</span>
|
|
<span id="L251"><span class="lineNum"> 251</span> : }</span>
|
|
<span id="L252"><span class="lineNum"> 252</span> : </span>
|
|
<span id="L253"><span class="lineNum"> 253</span> : /// @}</span>
|
|
<span id="L254"><span class="lineNum"> 254</span> : };</span>
|
|
<span id="L255"><span class="lineNum"> 255</span> : </span>
|
|
<span id="L256"><span class="lineNum"> 256</span> : template <typename T></span>
|
|
<span id="L257"><span class="lineNum"> 257</span> <span class="tlaGNC"> 36 : std::ostream& operator<<(std::ostream& out, ArrayRef<T> list) {</span></span>
|
|
<span id="L258"><span class="lineNum"> 258</span> <span class="tlaGNC"> 36 : int i = 0;</span></span>
|
|
<span id="L259"><span class="lineNum"> 259</span> <span class="tlaGNC"> 36 : out << "[";</span></span>
|
|
<span id="L260"><span class="lineNum"> 260</span> <span class="tlaGNC"> 108 : for (const auto& e : list) {</span></span>
|
|
<span id="L261"><span class="lineNum"> 261</span> <span class="tlaGNC"> 72 : if (i++ > 0)</span></span>
|
|
<span id="L262"><span class="lineNum"> 262</span> <span class="tlaGNC"> 36 : out << ", ";</span></span>
|
|
<span id="L263"><span class="lineNum"> 263</span> <span class="tlaGNC"> 72 : out << e;</span></span>
|
|
<span id="L264"><span class="lineNum"> 264</span> : }</span>
|
|
<span id="L265"><span class="lineNum"> 265</span> <span class="tlaGNC"> 36 : out << "]";</span></span>
|
|
<span id="L266"><span class="lineNum"> 266</span> <span class="tlaGNC"> 36 : return out;</span></span>
|
|
<span id="L267"><span class="lineNum"> 267</span> : }</span>
|
|
<span id="L268"><span class="lineNum"> 268</span> : </span>
|
|
<span id="L269"><span class="lineNum"> 269</span> : /// @name ArrayRef Convenience constructors</span>
|
|
<span id="L270"><span class="lineNum"> 270</span> : /// @{</span>
|
|
<span id="L271"><span class="lineNum"> 271</span> : </span>
|
|
<span id="L272"><span class="lineNum"> 272</span> : /// Construct an ArrayRef from a single element.</span>
|
|
<span id="L273"><span class="lineNum"> 273</span> : template <typename T></span>
|
|
<span id="L274"><span class="lineNum"> 274</span> : ArrayRef<T> makeArrayRef(const T& OneElt) {</span>
|
|
<span id="L275"><span class="lineNum"> 275</span> : return OneElt;</span>
|
|
<span id="L276"><span class="lineNum"> 276</span> : }</span>
|
|
<span id="L277"><span class="lineNum"> 277</span> : </span>
|
|
<span id="L278"><span class="lineNum"> 278</span> : /// Construct an ArrayRef from a pointer and length.</span>
|
|
<span id="L279"><span class="lineNum"> 279</span> : template <typename T></span>
|
|
<span id="L280"><span class="lineNum"> 280</span> : ArrayRef<T> makeArrayRef(const T* data, size_t length) {</span>
|
|
<span id="L281"><span class="lineNum"> 281</span> : return ArrayRef<T>(data, length);</span>
|
|
<span id="L282"><span class="lineNum"> 282</span> : }</span>
|
|
<span id="L283"><span class="lineNum"> 283</span> : </span>
|
|
<span id="L284"><span class="lineNum"> 284</span> : /// Construct an ArrayRef from a range.</span>
|
|
<span id="L285"><span class="lineNum"> 285</span> : template <typename T></span>
|
|
<span id="L286"><span class="lineNum"> 286</span> : ArrayRef<T> makeArrayRef(const T* begin, const T* end) {</span>
|
|
<span id="L287"><span class="lineNum"> 287</span> : return ArrayRef<T>(begin, end);</span>
|
|
<span id="L288"><span class="lineNum"> 288</span> : }</span>
|
|
<span id="L289"><span class="lineNum"> 289</span> : </span>
|
|
<span id="L290"><span class="lineNum"> 290</span> : /// Construct an ArrayRef from a SmallVector.</span>
|
|
<span id="L291"><span class="lineNum"> 291</span> : template <typename T></span>
|
|
<span id="L292"><span class="lineNum"> 292</span> : ArrayRef<T> makeArrayRef(const SmallVectorImpl<T>& Vec) {</span>
|
|
<span id="L293"><span class="lineNum"> 293</span> : return Vec;</span>
|
|
<span id="L294"><span class="lineNum"> 294</span> : }</span>
|
|
<span id="L295"><span class="lineNum"> 295</span> : </span>
|
|
<span id="L296"><span class="lineNum"> 296</span> : /// Construct an ArrayRef from a SmallVector.</span>
|
|
<span id="L297"><span class="lineNum"> 297</span> : template <typename T, unsigned N></span>
|
|
<span id="L298"><span class="lineNum"> 298</span> : ArrayRef<T> makeArrayRef(const SmallVector<T, N>& Vec) {</span>
|
|
<span id="L299"><span class="lineNum"> 299</span> : return Vec;</span>
|
|
<span id="L300"><span class="lineNum"> 300</span> : }</span>
|
|
<span id="L301"><span class="lineNum"> 301</span> : </span>
|
|
<span id="L302"><span class="lineNum"> 302</span> : /// Construct an ArrayRef from a std::vector.</span>
|
|
<span id="L303"><span class="lineNum"> 303</span> : template <typename T></span>
|
|
<span id="L304"><span class="lineNum"> 304</span> : ArrayRef<T> makeArrayRef(const std::vector<T>& Vec) {</span>
|
|
<span id="L305"><span class="lineNum"> 305</span> : return Vec;</span>
|
|
<span id="L306"><span class="lineNum"> 306</span> : }</span>
|
|
<span id="L307"><span class="lineNum"> 307</span> : </span>
|
|
<span id="L308"><span class="lineNum"> 308</span> : /// Construct an ArrayRef from a std::array.</span>
|
|
<span id="L309"><span class="lineNum"> 309</span> : template <typename T, std::size_t N></span>
|
|
<span id="L310"><span class="lineNum"> 310</span> : ArrayRef<T> makeArrayRef(const std::array<T, N>& Arr) {</span>
|
|
<span id="L311"><span class="lineNum"> 311</span> : return Arr;</span>
|
|
<span id="L312"><span class="lineNum"> 312</span> : }</span>
|
|
<span id="L313"><span class="lineNum"> 313</span> : </span>
|
|
<span id="L314"><span class="lineNum"> 314</span> : /// Construct an ArrayRef from an ArrayRef (no-op) (const)</span>
|
|
<span id="L315"><span class="lineNum"> 315</span> : template <typename T></span>
|
|
<span id="L316"><span class="lineNum"> 316</span> : ArrayRef<T> makeArrayRef(const ArrayRef<T>& Vec) {</span>
|
|
<span id="L317"><span class="lineNum"> 317</span> : return Vec;</span>
|
|
<span id="L318"><span class="lineNum"> 318</span> : }</span>
|
|
<span id="L319"><span class="lineNum"> 319</span> : </span>
|
|
<span id="L320"><span class="lineNum"> 320</span> : /// Construct an ArrayRef from an ArrayRef (no-op)</span>
|
|
<span id="L321"><span class="lineNum"> 321</span> : template <typename T></span>
|
|
<span id="L322"><span class="lineNum"> 322</span> : ArrayRef<T>& makeArrayRef(ArrayRef<T>& Vec) {</span>
|
|
<span id="L323"><span class="lineNum"> 323</span> : return Vec;</span>
|
|
<span id="L324"><span class="lineNum"> 324</span> : }</span>
|
|
<span id="L325"><span class="lineNum"> 325</span> : </span>
|
|
<span id="L326"><span class="lineNum"> 326</span> : /// Construct an ArrayRef from a C array.</span>
|
|
<span id="L327"><span class="lineNum"> 327</span> : template <typename T, size_t N></span>
|
|
<span id="L328"><span class="lineNum"> 328</span> : ArrayRef<T> makeArrayRef(const T (&Arr)[N]) {</span>
|
|
<span id="L329"><span class="lineNum"> 329</span> : return ArrayRef<T>(Arr);</span>
|
|
<span id="L330"><span class="lineNum"> 330</span> : }</span>
|
|
<span id="L331"><span class="lineNum"> 331</span> : </span>
|
|
<span id="L332"><span class="lineNum"> 332</span> : // WARNING: Template instantiation will NOT be willing to do an implicit</span>
|
|
<span id="L333"><span class="lineNum"> 333</span> : // conversions to get you to an c10::ArrayRef, which is why we need so</span>
|
|
<span id="L334"><span class="lineNum"> 334</span> : // many overloads.</span>
|
|
<span id="L335"><span class="lineNum"> 335</span> : </span>
|
|
<span id="L336"><span class="lineNum"> 336</span> : template <typename T></span>
|
|
<span id="L337"><span class="lineNum"> 337</span> : bool operator==(c10::ArrayRef<T> a1, c10::ArrayRef<T> a2) {</span>
|
|
<span id="L338"><span class="lineNum"> 338</span> : return a1.equals(a2);</span>
|
|
<span id="L339"><span class="lineNum"> 339</span> : }</span>
|
|
<span id="L340"><span class="lineNum"> 340</span> : </span>
|
|
<span id="L341"><span class="lineNum"> 341</span> : template <typename T></span>
|
|
<span id="L342"><span class="lineNum"> 342</span> : bool operator!=(c10::ArrayRef<T> a1, c10::ArrayRef<T> a2) {</span>
|
|
<span id="L343"><span class="lineNum"> 343</span> : return !a1.equals(a2);</span>
|
|
<span id="L344"><span class="lineNum"> 344</span> : }</span>
|
|
<span id="L345"><span class="lineNum"> 345</span> : </span>
|
|
<span id="L346"><span class="lineNum"> 346</span> : template <typename T></span>
|
|
<span id="L347"><span class="lineNum"> 347</span> : bool operator==(const std::vector<T>& a1, c10::ArrayRef<T> a2) {</span>
|
|
<span id="L348"><span class="lineNum"> 348</span> : return c10::ArrayRef<T>(a1).equals(a2);</span>
|
|
<span id="L349"><span class="lineNum"> 349</span> : }</span>
|
|
<span id="L350"><span class="lineNum"> 350</span> : </span>
|
|
<span id="L351"><span class="lineNum"> 351</span> : template <typename T></span>
|
|
<span id="L352"><span class="lineNum"> 352</span> : bool operator!=(const std::vector<T>& a1, c10::ArrayRef<T> a2) {</span>
|
|
<span id="L353"><span class="lineNum"> 353</span> : return !c10::ArrayRef<T>(a1).equals(a2);</span>
|
|
<span id="L354"><span class="lineNum"> 354</span> : }</span>
|
|
<span id="L355"><span class="lineNum"> 355</span> : </span>
|
|
<span id="L356"><span class="lineNum"> 356</span> : template <typename T></span>
|
|
<span id="L357"><span class="lineNum"> 357</span> : bool operator==(c10::ArrayRef<T> a1, const std::vector<T>& a2) {</span>
|
|
<span id="L358"><span class="lineNum"> 358</span> : return a1.equals(c10::ArrayRef<T>(a2));</span>
|
|
<span id="L359"><span class="lineNum"> 359</span> : }</span>
|
|
<span id="L360"><span class="lineNum"> 360</span> : </span>
|
|
<span id="L361"><span class="lineNum"> 361</span> : template <typename T></span>
|
|
<span id="L362"><span class="lineNum"> 362</span> : bool operator!=(c10::ArrayRef<T> a1, const std::vector<T>& a2) {</span>
|
|
<span id="L363"><span class="lineNum"> 363</span> : return !a1.equals(c10::ArrayRef<T>(a2));</span>
|
|
<span id="L364"><span class="lineNum"> 364</span> : }</span>
|
|
<span id="L365"><span class="lineNum"> 365</span> : </span>
|
|
<span id="L366"><span class="lineNum"> 366</span> : using IntArrayRef = ArrayRef<int64_t>;</span>
|
|
<span id="L367"><span class="lineNum"> 367</span> : </span>
|
|
<span id="L368"><span class="lineNum"> 368</span> : // This alias is deprecated because it doesn't make ownership</span>
|
|
<span id="L369"><span class="lineNum"> 369</span> : // semantics obvious. Use IntArrayRef instead!</span>
|
|
<span id="L370"><span class="lineNum"> 370</span> : C10_DEFINE_DEPRECATED_USING(IntList, ArrayRef<int64_t>)</span>
|
|
<span id="L371"><span class="lineNum"> 371</span> : </span>
|
|
<span id="L372"><span class="lineNum"> 372</span> : } // namespace c10</span>
|
|
</pre>
|
|
</td>
|
|
</tr>
|
|
</table>
|
|
<br>
|
|
|
|
<table width="100%" border=0 cellspacing=0 cellpadding=0>
|
|
<tr><td class="ruler"><img src="../../../../glass.png" width=3 height=3 alt=""></td></tr>
|
|
<tr><td class="versionInfo">Generated by: <a href="https://github.com//linux-test-project/lcov" target="_parent">LCOV version 2.0-1</a></td></tr>
|
|
</table>
|
|
<br>
|
|
|
|
</body>
|
|
</html>
|