cavis/libnd4j/include/graph/generated/array_generated.h

281 lines
6.4 KiB
C
Raw Normal View History

2021-02-10 14:40:04 +01:00
// automatically generated by the FlatBuffers compiler, do not modify
#ifndef FLATBUFFERS_GENERATED_ARRAY_ND4J_GRAPH_H_
#define FLATBUFFERS_GENERATED_ARRAY_ND4J_GRAPH_H_
#include "flatbuffers/flatbuffers.h"
namespace sd {
namespace graph {
struct FlatArray;
enum ByteOrder {
ByteOrder_LE = 0,
ByteOrder_BE = 1,
ByteOrder_MIN = ByteOrder_LE,
ByteOrder_MAX = ByteOrder_BE
};
inline const ByteOrder (&EnumValuesByteOrder())[2] {
static const ByteOrder values[] = {
ByteOrder_LE,
ByteOrder_BE
};
return values;
}
inline const char * const *EnumNamesByteOrder() {
static const char * const names[] = {
"LE",
"BE",
nullptr
};
return names;
}
inline const char *EnumNameByteOrder(ByteOrder e) {
const size_t index = static_cast<int>(e);
return EnumNamesByteOrder()[index];
}
enum DType {
DType_INHERIT = 0,
DType_BOOL = 1,
DType_FLOAT8 = 2,
DType_HALF = 3,
DType_HALF2 = 4,
DType_FLOAT = 5,
DType_DOUBLE = 6,
DType_INT8 = 7,
DType_INT16 = 8,
DType_INT32 = 9,
DType_INT64 = 10,
DType_UINT8 = 11,
DType_UINT16 = 12,
DType_UINT32 = 13,
DType_UINT64 = 14,
DType_QINT8 = 15,
DType_QINT16 = 16,
DType_BFLOAT16 = 17,
DType_UTF8 = 50,
DType_UTF16 = 51,
DType_UTF32 = 52,
DType_MIN = DType_INHERIT,
DType_MAX = DType_UTF32
};
inline const DType (&EnumValuesDType())[21] {
static const DType values[] = {
DType_INHERIT,
DType_BOOL,
DType_FLOAT8,
DType_HALF,
DType_HALF2,
DType_FLOAT,
DType_DOUBLE,
DType_INT8,
DType_INT16,
DType_INT32,
DType_INT64,
DType_UINT8,
DType_UINT16,
DType_UINT32,
DType_UINT64,
DType_QINT8,
DType_QINT16,
DType_BFLOAT16,
DType_UTF8,
DType_UTF16,
DType_UTF32
};
return values;
}
inline const char * const *EnumNamesDType() {
static const char * const names[] = {
"INHERIT",
"BOOL",
"FLOAT8",
"HALF",
"HALF2",
"FLOAT",
"DOUBLE",
"INT8",
"INT16",
"INT32",
"INT64",
"UINT8",
"UINT16",
"UINT32",
"UINT64",
"QINT8",
"QINT16",
"BFLOAT16",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"",
"UTF8",
"UTF16",
"UTF32",
nullptr
};
return names;
}
inline const char *EnumNameDType(DType e) {
const size_t index = static_cast<int>(e);
return EnumNamesDType()[index];
}
struct FlatArray FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table {
enum {
VT_SHAPE = 4,
VT_BUFFER = 6,
VT_DTYPE = 8,
VT_BYTEORDER = 10
};
const flatbuffers::Vector<int64_t> *shape() const {
return GetPointer<const flatbuffers::Vector<int64_t> *>(VT_SHAPE);
}
const flatbuffers::Vector<int8_t> *buffer() const {
return GetPointer<const flatbuffers::Vector<int8_t> *>(VT_BUFFER);
}
DType dtype() const {
return static_cast<DType>(GetField<int8_t>(VT_DTYPE, 0));
}
ByteOrder byteOrder() const {
return static_cast<ByteOrder>(GetField<int8_t>(VT_BYTEORDER, 0));
}
bool Verify(flatbuffers::Verifier &verifier) const {
return VerifyTableStart(verifier) &&
VerifyOffset(verifier, VT_SHAPE) &&
verifier.VerifyVector(shape()) &&
VerifyOffset(verifier, VT_BUFFER) &&
verifier.VerifyVector(buffer()) &&
VerifyField<int8_t>(verifier, VT_DTYPE) &&
VerifyField<int8_t>(verifier, VT_BYTEORDER) &&
verifier.EndTable();
}
};
struct FlatArrayBuilder {
flatbuffers::FlatBufferBuilder &fbb_;
flatbuffers::uoffset_t start_;
void add_shape(flatbuffers::Offset<flatbuffers::Vector<int64_t>> shape) {
fbb_.AddOffset(FlatArray::VT_SHAPE, shape);
}
void add_buffer(flatbuffers::Offset<flatbuffers::Vector<int8_t>> buffer) {
fbb_.AddOffset(FlatArray::VT_BUFFER, buffer);
}
void add_dtype(DType dtype) {
fbb_.AddElement<int8_t>(FlatArray::VT_DTYPE, static_cast<int8_t>(dtype), 0);
}
void add_byteOrder(ByteOrder byteOrder) {
fbb_.AddElement<int8_t>(FlatArray::VT_BYTEORDER, static_cast<int8_t>(byteOrder), 0);
}
explicit FlatArrayBuilder(flatbuffers::FlatBufferBuilder &_fbb)
: fbb_(_fbb) {
start_ = fbb_.StartTable();
}
FlatArrayBuilder &operator=(const FlatArrayBuilder &);
flatbuffers::Offset<FlatArray> Finish() {
const auto end = fbb_.EndTable(start_);
auto o = flatbuffers::Offset<FlatArray>(end);
return o;
}
};
inline flatbuffers::Offset<FlatArray> CreateFlatArray(
flatbuffers::FlatBufferBuilder &_fbb,
flatbuffers::Offset<flatbuffers::Vector<int64_t>> shape = 0,
flatbuffers::Offset<flatbuffers::Vector<int8_t>> buffer = 0,
DType dtype = DType_INHERIT,
ByteOrder byteOrder = ByteOrder_LE) {
FlatArrayBuilder builder_(_fbb);
builder_.add_buffer(buffer);
builder_.add_shape(shape);
builder_.add_byteOrder(byteOrder);
builder_.add_dtype(dtype);
return builder_.Finish();
}
inline flatbuffers::Offset<FlatArray> CreateFlatArrayDirect(
flatbuffers::FlatBufferBuilder &_fbb,
const std::vector<int64_t> *shape = nullptr,
const std::vector<int8_t> *buffer = nullptr,
DType dtype = DType_INHERIT,
ByteOrder byteOrder = ByteOrder_LE) {
return sd::graph::CreateFlatArray(
_fbb,
shape ? _fbb.CreateVector<int64_t>(*shape) : 0,
buffer ? _fbb.CreateVector<int8_t>(*buffer) : 0,
dtype,
byteOrder);
}
inline const sd::graph::FlatArray *GetFlatArray(const void *buf) {
return flatbuffers::GetRoot<sd::graph::FlatArray>(buf);
}
inline const sd::graph::FlatArray *GetSizePrefixedFlatArray(const void *buf) {
return flatbuffers::GetSizePrefixedRoot<sd::graph::FlatArray>(buf);
}
inline bool VerifyFlatArrayBuffer(
flatbuffers::Verifier &verifier) {
return verifier.VerifyBuffer<sd::graph::FlatArray>(nullptr);
}
inline bool VerifySizePrefixedFlatArrayBuffer(
flatbuffers::Verifier &verifier) {
return verifier.VerifySizePrefixedBuffer<sd::graph::FlatArray>(nullptr);
}
inline void FinishFlatArrayBuffer(
flatbuffers::FlatBufferBuilder &fbb,
flatbuffers::Offset<sd::graph::FlatArray> root) {
fbb.Finish(root);
}
inline void FinishSizePrefixedFlatArrayBuffer(
flatbuffers::FlatBufferBuilder &fbb,
flatbuffers::Offset<sd::graph::FlatArray> root) {
fbb.FinishSizePrefixed(root);
}
} // namespace graph
} // namespace sd
#endif // FLATBUFFERS_GENERATED_ARRAY_ND4J_GRAPH_H_