/src/arrow/cpp/src/generated/Tensor_generated.h
Line | Count | Source (jump to first uncovered line) |
1 | | // automatically generated by the FlatBuffers compiler, do not modify |
2 | | |
3 | | |
4 | | #ifndef FLATBUFFERS_GENERATED_TENSOR_ORG_APACHE_ARROW_FLATBUF_H_ |
5 | | #define FLATBUFFERS_GENERATED_TENSOR_ORG_APACHE_ARROW_FLATBUF_H_ |
6 | | |
7 | | #include "flatbuffers/flatbuffers.h" |
8 | | |
9 | | // Ensure the included flatbuffers.h is the same version as when this file was |
10 | | // generated, otherwise it may not be compatible. |
11 | | static_assert(FLATBUFFERS_VERSION_MAJOR == 24 && |
12 | | FLATBUFFERS_VERSION_MINOR == 3 && |
13 | | FLATBUFFERS_VERSION_REVISION == 6, |
14 | | "Non-compatible flatbuffers version included"); |
15 | | |
16 | | #include "Schema_generated.h" |
17 | | |
18 | | namespace org { |
19 | | namespace apache { |
20 | | namespace arrow { |
21 | | namespace flatbuf { |
22 | | |
23 | | struct TensorDim; |
24 | | struct TensorDimBuilder; |
25 | | |
26 | | struct Tensor; |
27 | | struct TensorBuilder; |
28 | | |
29 | | /// ---------------------------------------------------------------------- |
30 | | /// Data structures for dense tensors |
31 | | /// Shape data for a single axis in a tensor |
32 | | struct TensorDim FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table { |
33 | | typedef TensorDimBuilder Builder; |
34 | | enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE { |
35 | | VT_SIZE = 4, |
36 | | VT_NAME = 6 |
37 | | }; |
38 | | /// Length of dimension |
39 | 115k | int64_t size() const { |
40 | 115k | return GetField<int64_t>(VT_SIZE, 0); |
41 | 115k | } |
42 | | /// Name of the dimension, optional |
43 | 461k | const ::flatbuffers::String *name() const { |
44 | 461k | return GetPointer<const ::flatbuffers::String *>(VT_NAME); |
45 | 461k | } |
46 | 346k | bool Verify(::flatbuffers::Verifier &verifier) const { |
47 | 346k | return VerifyTableStart(verifier) && |
48 | 346k | VerifyField<int64_t>(verifier, VT_SIZE, 8) && |
49 | 346k | VerifyOffset(verifier, VT_NAME) && |
50 | 346k | verifier.VerifyString(name()) && |
51 | 346k | verifier.EndTable(); |
52 | 346k | } |
53 | | }; |
54 | | |
55 | | struct TensorDimBuilder { |
56 | | typedef TensorDim Table; |
57 | | ::flatbuffers::FlatBufferBuilder &fbb_; |
58 | | ::flatbuffers::uoffset_t start_; |
59 | 0 | void add_size(int64_t size) { |
60 | 0 | fbb_.AddElement<int64_t>(TensorDim::VT_SIZE, size, 0); |
61 | 0 | } |
62 | 0 | void add_name(::flatbuffers::Offset<::flatbuffers::String> name) { |
63 | 0 | fbb_.AddOffset(TensorDim::VT_NAME, name); |
64 | 0 | } |
65 | | explicit TensorDimBuilder(::flatbuffers::FlatBufferBuilder &_fbb) |
66 | 0 | : fbb_(_fbb) { |
67 | 0 | start_ = fbb_.StartTable(); |
68 | 0 | } |
69 | 0 | ::flatbuffers::Offset<TensorDim> Finish() { |
70 | 0 | const auto end = fbb_.EndTable(start_); |
71 | 0 | auto o = ::flatbuffers::Offset<TensorDim>(end); |
72 | 0 | return o; |
73 | 0 | } |
74 | | }; |
75 | | |
76 | | inline ::flatbuffers::Offset<TensorDim> CreateTensorDim( |
77 | | ::flatbuffers::FlatBufferBuilder &_fbb, |
78 | | int64_t size = 0, |
79 | 0 | ::flatbuffers::Offset<::flatbuffers::String> name = 0) { |
80 | 0 | TensorDimBuilder builder_(_fbb); |
81 | 0 | builder_.add_size(size); |
82 | 0 | builder_.add_name(name); |
83 | 0 | return builder_.Finish(); |
84 | 0 | } |
85 | | |
86 | | inline ::flatbuffers::Offset<TensorDim> CreateTensorDimDirect( |
87 | | ::flatbuffers::FlatBufferBuilder &_fbb, |
88 | | int64_t size = 0, |
89 | 0 | const char *name = nullptr) { |
90 | 0 | auto name__ = name ? _fbb.CreateString(name) : 0; |
91 | 0 | return org::apache::arrow::flatbuf::CreateTensorDim( |
92 | 0 | _fbb, |
93 | 0 | size, |
94 | 0 | name__); |
95 | 0 | } |
96 | | |
97 | | struct Tensor FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table { |
98 | | typedef TensorBuilder Builder; |
99 | | enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE { |
100 | | VT_TYPE_TYPE = 4, |
101 | | VT_TYPE = 6, |
102 | | VT_SHAPE = 8, |
103 | | VT_STRIDES = 10, |
104 | | VT_DATA = 12 |
105 | | }; |
106 | 51.1k | org::apache::arrow::flatbuf::Type type_type() const { |
107 | 51.1k | return static_cast<org::apache::arrow::flatbuf::Type>(GetField<uint8_t>(VT_TYPE_TYPE, 0)); |
108 | 51.1k | } |
109 | | /// The type of data contained in a value cell. Currently only fixed-width |
110 | | /// value types are supported, no strings or nested types |
111 | 51.1k | const void *type() const { |
112 | 51.1k | return GetPointer<const void *>(VT_TYPE); |
113 | 51.1k | } |
114 | | template<typename T> const T *type_as() const; |
115 | 0 | const org::apache::arrow::flatbuf::Null *type_as_Null() const { |
116 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_Null ? static_cast<const org::apache::arrow::flatbuf::Null *>(type()) : nullptr; |
117 | 0 | } |
118 | 0 | const org::apache::arrow::flatbuf::Int *type_as_Int() const { |
119 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_Int ? static_cast<const org::apache::arrow::flatbuf::Int *>(type()) : nullptr; |
120 | 0 | } |
121 | 0 | const org::apache::arrow::flatbuf::FloatingPoint *type_as_FloatingPoint() const { |
122 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_FloatingPoint ? static_cast<const org::apache::arrow::flatbuf::FloatingPoint *>(type()) : nullptr; |
123 | 0 | } |
124 | 0 | const org::apache::arrow::flatbuf::Binary *type_as_Binary() const { |
125 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_Binary ? static_cast<const org::apache::arrow::flatbuf::Binary *>(type()) : nullptr; |
126 | 0 | } |
127 | 0 | const org::apache::arrow::flatbuf::Utf8 *type_as_Utf8() const { |
128 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_Utf8 ? static_cast<const org::apache::arrow::flatbuf::Utf8 *>(type()) : nullptr; |
129 | 0 | } |
130 | 0 | const org::apache::arrow::flatbuf::Bool *type_as_Bool() const { |
131 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_Bool ? static_cast<const org::apache::arrow::flatbuf::Bool *>(type()) : nullptr; |
132 | 0 | } |
133 | 0 | const org::apache::arrow::flatbuf::Decimal *type_as_Decimal() const { |
134 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_Decimal ? static_cast<const org::apache::arrow::flatbuf::Decimal *>(type()) : nullptr; |
135 | 0 | } |
136 | 0 | const org::apache::arrow::flatbuf::Date *type_as_Date() const { |
137 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_Date ? static_cast<const org::apache::arrow::flatbuf::Date *>(type()) : nullptr; |
138 | 0 | } |
139 | 0 | const org::apache::arrow::flatbuf::Time *type_as_Time() const { |
140 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_Time ? static_cast<const org::apache::arrow::flatbuf::Time *>(type()) : nullptr; |
141 | 0 | } |
142 | 0 | const org::apache::arrow::flatbuf::Timestamp *type_as_Timestamp() const { |
143 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_Timestamp ? static_cast<const org::apache::arrow::flatbuf::Timestamp *>(type()) : nullptr; |
144 | 0 | } |
145 | 0 | const org::apache::arrow::flatbuf::Interval *type_as_Interval() const { |
146 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_Interval ? static_cast<const org::apache::arrow::flatbuf::Interval *>(type()) : nullptr; |
147 | 0 | } |
148 | 0 | const org::apache::arrow::flatbuf::List *type_as_List() const { |
149 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_List ? static_cast<const org::apache::arrow::flatbuf::List *>(type()) : nullptr; |
150 | 0 | } |
151 | 0 | const org::apache::arrow::flatbuf::Struct_ *type_as_Struct_() const { |
152 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_Struct_ ? static_cast<const org::apache::arrow::flatbuf::Struct_ *>(type()) : nullptr; |
153 | 0 | } |
154 | 0 | const org::apache::arrow::flatbuf::Union *type_as_Union() const { |
155 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_Union ? static_cast<const org::apache::arrow::flatbuf::Union *>(type()) : nullptr; |
156 | 0 | } |
157 | 0 | const org::apache::arrow::flatbuf::FixedSizeBinary *type_as_FixedSizeBinary() const { |
158 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_FixedSizeBinary ? static_cast<const org::apache::arrow::flatbuf::FixedSizeBinary *>(type()) : nullptr; |
159 | 0 | } |
160 | 0 | const org::apache::arrow::flatbuf::FixedSizeList *type_as_FixedSizeList() const { |
161 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_FixedSizeList ? static_cast<const org::apache::arrow::flatbuf::FixedSizeList *>(type()) : nullptr; |
162 | 0 | } |
163 | 0 | const org::apache::arrow::flatbuf::Map *type_as_Map() const { |
164 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_Map ? static_cast<const org::apache::arrow::flatbuf::Map *>(type()) : nullptr; |
165 | 0 | } |
166 | 0 | const org::apache::arrow::flatbuf::Duration *type_as_Duration() const { |
167 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_Duration ? static_cast<const org::apache::arrow::flatbuf::Duration *>(type()) : nullptr; |
168 | 0 | } |
169 | 0 | const org::apache::arrow::flatbuf::LargeBinary *type_as_LargeBinary() const { |
170 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_LargeBinary ? static_cast<const org::apache::arrow::flatbuf::LargeBinary *>(type()) : nullptr; |
171 | 0 | } |
172 | 0 | const org::apache::arrow::flatbuf::LargeUtf8 *type_as_LargeUtf8() const { |
173 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_LargeUtf8 ? static_cast<const org::apache::arrow::flatbuf::LargeUtf8 *>(type()) : nullptr; |
174 | 0 | } |
175 | 0 | const org::apache::arrow::flatbuf::LargeList *type_as_LargeList() const { |
176 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_LargeList ? static_cast<const org::apache::arrow::flatbuf::LargeList *>(type()) : nullptr; |
177 | 0 | } |
178 | 0 | const org::apache::arrow::flatbuf::RunEndEncoded *type_as_RunEndEncoded() const { |
179 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_RunEndEncoded ? static_cast<const org::apache::arrow::flatbuf::RunEndEncoded *>(type()) : nullptr; |
180 | 0 | } |
181 | 0 | const org::apache::arrow::flatbuf::BinaryView *type_as_BinaryView() const { |
182 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_BinaryView ? static_cast<const org::apache::arrow::flatbuf::BinaryView *>(type()) : nullptr; |
183 | 0 | } |
184 | 0 | const org::apache::arrow::flatbuf::Utf8View *type_as_Utf8View() const { |
185 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_Utf8View ? static_cast<const org::apache::arrow::flatbuf::Utf8View *>(type()) : nullptr; |
186 | 0 | } |
187 | 0 | const org::apache::arrow::flatbuf::ListView *type_as_ListView() const { |
188 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_ListView ? static_cast<const org::apache::arrow::flatbuf::ListView *>(type()) : nullptr; |
189 | 0 | } |
190 | 0 | const org::apache::arrow::flatbuf::LargeListView *type_as_LargeListView() const { |
191 | 0 | return type_type() == org::apache::arrow::flatbuf::Type_LargeListView ? static_cast<const org::apache::arrow::flatbuf::LargeListView *>(type()) : nullptr; |
192 | 0 | } |
193 | | /// The dimensions of the tensor, optionally named |
194 | 203k | const ::flatbuffers::Vector<::flatbuffers::Offset<org::apache::arrow::flatbuf::TensorDim>> *shape() const { |
195 | 203k | return GetPointer<const ::flatbuffers::Vector<::flatbuffers::Offset<org::apache::arrow::flatbuf::TensorDim>> *>(VT_SHAPE); |
196 | 203k | } |
197 | | /// Non-negative byte offsets to advance one value cell along each dimension |
198 | | /// If omitted, default to row-major order (C-like). |
199 | 64.5k | const ::flatbuffers::Vector<int64_t> *strides() const { |
200 | 64.5k | return GetPointer<const ::flatbuffers::Vector<int64_t> *>(VT_STRIDES); |
201 | 64.5k | } |
202 | | /// The location and size of the tensor's data |
203 | 0 | const org::apache::arrow::flatbuf::Buffer *data() const { |
204 | 0 | return GetStruct<const org::apache::arrow::flatbuf::Buffer *>(VT_DATA); |
205 | 0 | } |
206 | 38.9k | bool Verify(::flatbuffers::Verifier &verifier) const { |
207 | 38.9k | return VerifyTableStart(verifier) && |
208 | 38.9k | VerifyField<uint8_t>(verifier, VT_TYPE_TYPE, 1) && |
209 | 38.9k | VerifyOffsetRequired(verifier, VT_TYPE) && |
210 | 38.9k | VerifyType(verifier, type(), type_type()) && |
211 | 38.9k | VerifyOffsetRequired(verifier, VT_SHAPE) && |
212 | 38.9k | verifier.VerifyVector(shape()) && |
213 | 38.9k | verifier.VerifyVectorOfTables(shape()) && |
214 | 38.9k | VerifyOffset(verifier, VT_STRIDES) && |
215 | 38.9k | verifier.VerifyVector(strides()) && |
216 | 38.9k | VerifyFieldRequired<org::apache::arrow::flatbuf::Buffer>(verifier, VT_DATA, 8) && |
217 | 38.9k | verifier.EndTable(); |
218 | 38.9k | } |
219 | | }; |
220 | | |
221 | 0 | template<> inline const org::apache::arrow::flatbuf::Null *Tensor::type_as<org::apache::arrow::flatbuf::Null>() const { |
222 | 0 | return type_as_Null(); |
223 | 0 | } |
224 | | |
225 | 0 | template<> inline const org::apache::arrow::flatbuf::Int *Tensor::type_as<org::apache::arrow::flatbuf::Int>() const { |
226 | 0 | return type_as_Int(); |
227 | 0 | } |
228 | | |
229 | 0 | template<> inline const org::apache::arrow::flatbuf::FloatingPoint *Tensor::type_as<org::apache::arrow::flatbuf::FloatingPoint>() const { |
230 | 0 | return type_as_FloatingPoint(); |
231 | 0 | } |
232 | | |
233 | 0 | template<> inline const org::apache::arrow::flatbuf::Binary *Tensor::type_as<org::apache::arrow::flatbuf::Binary>() const { |
234 | 0 | return type_as_Binary(); |
235 | 0 | } |
236 | | |
237 | 0 | template<> inline const org::apache::arrow::flatbuf::Utf8 *Tensor::type_as<org::apache::arrow::flatbuf::Utf8>() const { |
238 | 0 | return type_as_Utf8(); |
239 | 0 | } |
240 | | |
241 | 0 | template<> inline const org::apache::arrow::flatbuf::Bool *Tensor::type_as<org::apache::arrow::flatbuf::Bool>() const { |
242 | 0 | return type_as_Bool(); |
243 | 0 | } |
244 | | |
245 | 0 | template<> inline const org::apache::arrow::flatbuf::Decimal *Tensor::type_as<org::apache::arrow::flatbuf::Decimal>() const { |
246 | 0 | return type_as_Decimal(); |
247 | 0 | } |
248 | | |
249 | 0 | template<> inline const org::apache::arrow::flatbuf::Date *Tensor::type_as<org::apache::arrow::flatbuf::Date>() const { |
250 | 0 | return type_as_Date(); |
251 | 0 | } |
252 | | |
253 | 0 | template<> inline const org::apache::arrow::flatbuf::Time *Tensor::type_as<org::apache::arrow::flatbuf::Time>() const { |
254 | 0 | return type_as_Time(); |
255 | 0 | } |
256 | | |
257 | 0 | template<> inline const org::apache::arrow::flatbuf::Timestamp *Tensor::type_as<org::apache::arrow::flatbuf::Timestamp>() const { |
258 | 0 | return type_as_Timestamp(); |
259 | 0 | } |
260 | | |
261 | 0 | template<> inline const org::apache::arrow::flatbuf::Interval *Tensor::type_as<org::apache::arrow::flatbuf::Interval>() const { |
262 | 0 | return type_as_Interval(); |
263 | 0 | } |
264 | | |
265 | 0 | template<> inline const org::apache::arrow::flatbuf::List *Tensor::type_as<org::apache::arrow::flatbuf::List>() const { |
266 | 0 | return type_as_List(); |
267 | 0 | } |
268 | | |
269 | 0 | template<> inline const org::apache::arrow::flatbuf::Struct_ *Tensor::type_as<org::apache::arrow::flatbuf::Struct_>() const { |
270 | 0 | return type_as_Struct_(); |
271 | 0 | } |
272 | | |
273 | 0 | template<> inline const org::apache::arrow::flatbuf::Union *Tensor::type_as<org::apache::arrow::flatbuf::Union>() const { |
274 | 0 | return type_as_Union(); |
275 | 0 | } |
276 | | |
277 | 0 | template<> inline const org::apache::arrow::flatbuf::FixedSizeBinary *Tensor::type_as<org::apache::arrow::flatbuf::FixedSizeBinary>() const { |
278 | 0 | return type_as_FixedSizeBinary(); |
279 | 0 | } |
280 | | |
281 | 0 | template<> inline const org::apache::arrow::flatbuf::FixedSizeList *Tensor::type_as<org::apache::arrow::flatbuf::FixedSizeList>() const { |
282 | 0 | return type_as_FixedSizeList(); |
283 | 0 | } |
284 | | |
285 | 0 | template<> inline const org::apache::arrow::flatbuf::Map *Tensor::type_as<org::apache::arrow::flatbuf::Map>() const { |
286 | 0 | return type_as_Map(); |
287 | 0 | } |
288 | | |
289 | 0 | template<> inline const org::apache::arrow::flatbuf::Duration *Tensor::type_as<org::apache::arrow::flatbuf::Duration>() const { |
290 | 0 | return type_as_Duration(); |
291 | 0 | } |
292 | | |
293 | 0 | template<> inline const org::apache::arrow::flatbuf::LargeBinary *Tensor::type_as<org::apache::arrow::flatbuf::LargeBinary>() const { |
294 | 0 | return type_as_LargeBinary(); |
295 | 0 | } |
296 | | |
297 | 0 | template<> inline const org::apache::arrow::flatbuf::LargeUtf8 *Tensor::type_as<org::apache::arrow::flatbuf::LargeUtf8>() const { |
298 | 0 | return type_as_LargeUtf8(); |
299 | 0 | } |
300 | | |
301 | 0 | template<> inline const org::apache::arrow::flatbuf::LargeList *Tensor::type_as<org::apache::arrow::flatbuf::LargeList>() const { |
302 | 0 | return type_as_LargeList(); |
303 | 0 | } |
304 | | |
305 | 0 | template<> inline const org::apache::arrow::flatbuf::RunEndEncoded *Tensor::type_as<org::apache::arrow::flatbuf::RunEndEncoded>() const { |
306 | 0 | return type_as_RunEndEncoded(); |
307 | 0 | } |
308 | | |
309 | 0 | template<> inline const org::apache::arrow::flatbuf::BinaryView *Tensor::type_as<org::apache::arrow::flatbuf::BinaryView>() const { |
310 | 0 | return type_as_BinaryView(); |
311 | 0 | } |
312 | | |
313 | 0 | template<> inline const org::apache::arrow::flatbuf::Utf8View *Tensor::type_as<org::apache::arrow::flatbuf::Utf8View>() const { |
314 | 0 | return type_as_Utf8View(); |
315 | 0 | } |
316 | | |
317 | 0 | template<> inline const org::apache::arrow::flatbuf::ListView *Tensor::type_as<org::apache::arrow::flatbuf::ListView>() const { |
318 | 0 | return type_as_ListView(); |
319 | 0 | } |
320 | | |
321 | 0 | template<> inline const org::apache::arrow::flatbuf::LargeListView *Tensor::type_as<org::apache::arrow::flatbuf::LargeListView>() const { |
322 | 0 | return type_as_LargeListView(); |
323 | 0 | } |
324 | | |
325 | | struct TensorBuilder { |
326 | | typedef Tensor Table; |
327 | | ::flatbuffers::FlatBufferBuilder &fbb_; |
328 | | ::flatbuffers::uoffset_t start_; |
329 | 0 | void add_type_type(org::apache::arrow::flatbuf::Type type_type) { |
330 | 0 | fbb_.AddElement<uint8_t>(Tensor::VT_TYPE_TYPE, static_cast<uint8_t>(type_type), 0); |
331 | 0 | } |
332 | 0 | void add_type(::flatbuffers::Offset<void> type) { |
333 | 0 | fbb_.AddOffset(Tensor::VT_TYPE, type); |
334 | 0 | } |
335 | 0 | void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<org::apache::arrow::flatbuf::TensorDim>>> shape) { |
336 | 0 | fbb_.AddOffset(Tensor::VT_SHAPE, shape); |
337 | 0 | } |
338 | 0 | void add_strides(::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> strides) { |
339 | 0 | fbb_.AddOffset(Tensor::VT_STRIDES, strides); |
340 | 0 | } |
341 | 0 | void add_data(const org::apache::arrow::flatbuf::Buffer *data) { |
342 | 0 | fbb_.AddStruct(Tensor::VT_DATA, data); |
343 | 0 | } |
344 | | explicit TensorBuilder(::flatbuffers::FlatBufferBuilder &_fbb) |
345 | 0 | : fbb_(_fbb) { |
346 | 0 | start_ = fbb_.StartTable(); |
347 | 0 | } |
348 | 0 | ::flatbuffers::Offset<Tensor> Finish() { |
349 | 0 | const auto end = fbb_.EndTable(start_); |
350 | 0 | auto o = ::flatbuffers::Offset<Tensor>(end); |
351 | 0 | fbb_.Required(o, Tensor::VT_TYPE); |
352 | 0 | fbb_.Required(o, Tensor::VT_SHAPE); |
353 | 0 | fbb_.Required(o, Tensor::VT_DATA); |
354 | 0 | return o; |
355 | 0 | } |
356 | | }; |
357 | | |
358 | | inline ::flatbuffers::Offset<Tensor> CreateTensor( |
359 | | ::flatbuffers::FlatBufferBuilder &_fbb, |
360 | | org::apache::arrow::flatbuf::Type type_type = org::apache::arrow::flatbuf::Type_NONE, |
361 | | ::flatbuffers::Offset<void> type = 0, |
362 | | ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<org::apache::arrow::flatbuf::TensorDim>>> shape = 0, |
363 | | ::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> strides = 0, |
364 | 0 | const org::apache::arrow::flatbuf::Buffer *data = nullptr) { |
365 | 0 | TensorBuilder builder_(_fbb); |
366 | 0 | builder_.add_data(data); |
367 | 0 | builder_.add_strides(strides); |
368 | 0 | builder_.add_shape(shape); |
369 | 0 | builder_.add_type(type); |
370 | 0 | builder_.add_type_type(type_type); |
371 | 0 | return builder_.Finish(); |
372 | 0 | } |
373 | | |
374 | | inline ::flatbuffers::Offset<Tensor> CreateTensorDirect( |
375 | | ::flatbuffers::FlatBufferBuilder &_fbb, |
376 | | org::apache::arrow::flatbuf::Type type_type = org::apache::arrow::flatbuf::Type_NONE, |
377 | | ::flatbuffers::Offset<void> type = 0, |
378 | | const std::vector<::flatbuffers::Offset<org::apache::arrow::flatbuf::TensorDim>> *shape = nullptr, |
379 | | const std::vector<int64_t> *strides = nullptr, |
380 | 0 | const org::apache::arrow::flatbuf::Buffer *data = nullptr) { |
381 | 0 | auto shape__ = shape ? _fbb.CreateVector<::flatbuffers::Offset<org::apache::arrow::flatbuf::TensorDim>>(*shape) : 0; |
382 | 0 | auto strides__ = strides ? _fbb.CreateVector<int64_t>(*strides) : 0; |
383 | 0 | return org::apache::arrow::flatbuf::CreateTensor( |
384 | 0 | _fbb, |
385 | 0 | type_type, |
386 | 0 | type, |
387 | 0 | shape__, |
388 | 0 | strides__, |
389 | 0 | data); |
390 | 0 | } |
391 | | |
392 | 0 | inline const org::apache::arrow::flatbuf::Tensor *GetTensor(const void *buf) { |
393 | 0 | return ::flatbuffers::GetRoot<org::apache::arrow::flatbuf::Tensor>(buf); |
394 | 0 | } |
395 | | |
396 | 0 | inline const org::apache::arrow::flatbuf::Tensor *GetSizePrefixedTensor(const void *buf) { |
397 | 0 | return ::flatbuffers::GetSizePrefixedRoot<org::apache::arrow::flatbuf::Tensor>(buf); |
398 | 0 | } |
399 | | |
400 | | inline bool VerifyTensorBuffer( |
401 | 0 | ::flatbuffers::Verifier &verifier) { |
402 | 0 | return verifier.VerifyBuffer<org::apache::arrow::flatbuf::Tensor>(nullptr); |
403 | 0 | } |
404 | | |
405 | | inline bool VerifySizePrefixedTensorBuffer( |
406 | 0 | ::flatbuffers::Verifier &verifier) { |
407 | 0 | return verifier.VerifySizePrefixedBuffer<org::apache::arrow::flatbuf::Tensor>(nullptr); |
408 | 0 | } |
409 | | |
410 | | inline void FinishTensorBuffer( |
411 | | ::flatbuffers::FlatBufferBuilder &fbb, |
412 | 0 | ::flatbuffers::Offset<org::apache::arrow::flatbuf::Tensor> root) { |
413 | 0 | fbb.Finish(root); |
414 | 0 | } |
415 | | |
416 | | inline void FinishSizePrefixedTensorBuffer( |
417 | | ::flatbuffers::FlatBufferBuilder &fbb, |
418 | 0 | ::flatbuffers::Offset<org::apache::arrow::flatbuf::Tensor> root) { |
419 | 0 | fbb.FinishSizePrefixed(root); |
420 | 0 | } |
421 | | |
422 | | } // namespace flatbuf |
423 | | } // namespace arrow |
424 | | } // namespace apache |
425 | | } // namespace org |
426 | | |
427 | | #endif // FLATBUFFERS_GENERATED_TENSOR_ORG_APACHE_ARROW_FLATBUF_H_ |