// automatically generated by the FlatBuffers compiler, do not modify #ifndef FLATBUFFERS_GENERATED_VARIABLE_ND4J_GRAPH_H_ #define FLATBUFFERS_GENERATED_VARIABLE_ND4J_GRAPH_H_ #include "flatbuffers/flatbuffers.h" #include "array_generated.h" #include "utils_generated.h" namespace nd4j { namespace graph { struct FlatVariable; enum VarType { VarType_VARIABLE = 0, VarType_CONSTANT = 1, VarType_ARRAY = 2, VarType_PLACEHOLDER = 3, VarType_MIN = VarType_VARIABLE, VarType_MAX = VarType_PLACEHOLDER }; inline const VarType (&EnumValuesVarType())[4] { static const VarType values[] = { VarType_VARIABLE, VarType_CONSTANT, VarType_ARRAY, VarType_PLACEHOLDER }; return values; } inline const char * const *EnumNamesVarType() { static const char * const names[] = { "VARIABLE", "CONSTANT", "ARRAY", "PLACEHOLDER", nullptr }; return names; } inline const char *EnumNameVarType(VarType e) { const size_t index = static_cast(e); return EnumNamesVarType()[index]; } struct FlatVariable FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table { enum { VT_ID = 4, VT_NAME = 6, VT_DTYPE = 8, VT_SHAPE = 10, VT_NDARRAY = 12, VT_DEVICE = 14, VT_VARIABLETYPE = 16, VT_CONTROLDEPS = 18, VT_CONTROLDEPFOROP = 20, VT_CONTROLDEPSFORVAR = 22 }; const IntPair *id() const { return GetPointer(VT_ID); } const flatbuffers::String *name() const { return GetPointer(VT_NAME); } DType dtype() const { return static_cast(GetField(VT_DTYPE, 0)); } const flatbuffers::Vector *shape() const { return GetPointer *>(VT_SHAPE); } const FlatArray *ndarray() const { return GetPointer(VT_NDARRAY); } int32_t device() const { return GetField(VT_DEVICE, 0); } VarType variabletype() const { return static_cast(GetField(VT_VARIABLETYPE, 0)); } const flatbuffers::Vector> *controlDeps() const { return GetPointer> *>(VT_CONTROLDEPS); } const flatbuffers::Vector> *controlDepForOp() const { return GetPointer> *>(VT_CONTROLDEPFOROP); } const flatbuffers::Vector> *controlDepsForVar() const { return GetPointer> *>(VT_CONTROLDEPSFORVAR); } bool Verify(flatbuffers::Verifier &verifier) const { return VerifyTableStart(verifier) && VerifyOffset(verifier, VT_ID) && verifier.VerifyTable(id()) && VerifyOffset(verifier, VT_NAME) && verifier.VerifyString(name()) && VerifyField(verifier, VT_DTYPE) && VerifyOffset(verifier, VT_SHAPE) && verifier.VerifyVector(shape()) && VerifyOffset(verifier, VT_NDARRAY) && verifier.VerifyTable(ndarray()) && VerifyField(verifier, VT_DEVICE) && VerifyField(verifier, VT_VARIABLETYPE) && VerifyOffset(verifier, VT_CONTROLDEPS) && verifier.VerifyVector(controlDeps()) && verifier.VerifyVectorOfStrings(controlDeps()) && VerifyOffset(verifier, VT_CONTROLDEPFOROP) && verifier.VerifyVector(controlDepForOp()) && verifier.VerifyVectorOfStrings(controlDepForOp()) && VerifyOffset(verifier, VT_CONTROLDEPSFORVAR) && verifier.VerifyVector(controlDepsForVar()) && verifier.VerifyVectorOfStrings(controlDepsForVar()) && verifier.EndTable(); } }; struct FlatVariableBuilder { flatbuffers::FlatBufferBuilder &fbb_; flatbuffers::uoffset_t start_; void add_id(flatbuffers::Offset id) { fbb_.AddOffset(FlatVariable::VT_ID, id); } void add_name(flatbuffers::Offset name) { fbb_.AddOffset(FlatVariable::VT_NAME, name); } void add_dtype(DType dtype) { fbb_.AddElement(FlatVariable::VT_DTYPE, static_cast(dtype), 0); } void add_shape(flatbuffers::Offset> shape) { fbb_.AddOffset(FlatVariable::VT_SHAPE, shape); } void add_ndarray(flatbuffers::Offset ndarray) { fbb_.AddOffset(FlatVariable::VT_NDARRAY, ndarray); } void add_device(int32_t device) { fbb_.AddElement(FlatVariable::VT_DEVICE, device, 0); } void add_variabletype(VarType variabletype) { fbb_.AddElement(FlatVariable::VT_VARIABLETYPE, static_cast(variabletype), 0); } void add_controlDeps(flatbuffers::Offset>> controlDeps) { fbb_.AddOffset(FlatVariable::VT_CONTROLDEPS, controlDeps); } void add_controlDepForOp(flatbuffers::Offset>> controlDepForOp) { fbb_.AddOffset(FlatVariable::VT_CONTROLDEPFOROP, controlDepForOp); } void add_controlDepsForVar(flatbuffers::Offset>> controlDepsForVar) { fbb_.AddOffset(FlatVariable::VT_CONTROLDEPSFORVAR, controlDepsForVar); } explicit FlatVariableBuilder(flatbuffers::FlatBufferBuilder &_fbb) : fbb_(_fbb) { start_ = fbb_.StartTable(); } FlatVariableBuilder &operator=(const FlatVariableBuilder &); flatbuffers::Offset Finish() { const auto end = fbb_.EndTable(start_); auto o = flatbuffers::Offset(end); return o; } }; inline flatbuffers::Offset CreateFlatVariable( flatbuffers::FlatBufferBuilder &_fbb, flatbuffers::Offset id = 0, flatbuffers::Offset name = 0, DType dtype = DType_INHERIT, flatbuffers::Offset> shape = 0, flatbuffers::Offset ndarray = 0, int32_t device = 0, VarType variabletype = VarType_VARIABLE, flatbuffers::Offset>> controlDeps = 0, flatbuffers::Offset>> controlDepForOp = 0, flatbuffers::Offset>> controlDepsForVar = 0) { FlatVariableBuilder builder_(_fbb); builder_.add_controlDepsForVar(controlDepsForVar); builder_.add_controlDepForOp(controlDepForOp); builder_.add_controlDeps(controlDeps); builder_.add_device(device); builder_.add_ndarray(ndarray); builder_.add_shape(shape); builder_.add_name(name); builder_.add_id(id); builder_.add_variabletype(variabletype); builder_.add_dtype(dtype); return builder_.Finish(); } inline flatbuffers::Offset CreateFlatVariableDirect( flatbuffers::FlatBufferBuilder &_fbb, flatbuffers::Offset id = 0, const char *name = nullptr, DType dtype = DType_INHERIT, const std::vector *shape = nullptr, flatbuffers::Offset ndarray = 0, int32_t device = 0, VarType variabletype = VarType_VARIABLE, const std::vector> *controlDeps = nullptr, const std::vector> *controlDepForOp = nullptr, const std::vector> *controlDepsForVar = nullptr) { return nd4j::graph::CreateFlatVariable( _fbb, id, name ? _fbb.CreateString(name) : 0, dtype, shape ? _fbb.CreateVector(*shape) : 0, ndarray, device, variabletype, controlDeps ? _fbb.CreateVector>(*controlDeps) : 0, controlDepForOp ? _fbb.CreateVector>(*controlDepForOp) : 0, controlDepsForVar ? _fbb.CreateVector>(*controlDepsForVar) : 0); } inline const nd4j::graph::FlatVariable *GetFlatVariable(const void *buf) { return flatbuffers::GetRoot(buf); } inline const nd4j::graph::FlatVariable *GetSizePrefixedFlatVariable(const void *buf) { return flatbuffers::GetSizePrefixedRoot(buf); } inline bool VerifyFlatVariableBuffer( flatbuffers::Verifier &verifier) { return verifier.VerifyBuffer(nullptr); } inline bool VerifySizePrefixedFlatVariableBuffer( flatbuffers::Verifier &verifier) { return verifier.VerifySizePrefixedBuffer(nullptr); } inline void FinishFlatVariableBuffer( flatbuffers::FlatBufferBuilder &fbb, flatbuffers::Offset root) { fbb.Finish(root); } inline void FinishSizePrefixedFlatVariableBuffer( flatbuffers::FlatBufferBuilder &fbb, flatbuffers::Offset root) { fbb.FinishSizePrefixed(root); } } // namespace graph } // namespace nd4j #endif // FLATBUFFERS_GENERATED_VARIABLE_ND4J_GRAPH_H_