aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/schema
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-09-17 01:59:43 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-17 02:04:10 -0700
commit0827dcb82b0bf3d8d543cef1d3a17d330f183848 (patch)
tree2c55ba9c98ac6844a1528dfc01b57c32b5717d27 /tensorflow/contrib/lite/schema
parentf1d42c8967410db1e08c0b6d62dc1fc4844165a8 (diff)
Add fill to schema.
PiperOrigin-RevId: 213234759
Diffstat (limited to 'tensorflow/contrib/lite/schema')
-rw-r--r--tensorflow/contrib/lite/schema/schema.fbs5
-rwxr-xr-xtensorflow/contrib/lite/schema/schema_generated.h124
2 files changed, 123 insertions, 6 deletions
diff --git a/tensorflow/contrib/lite/schema/schema.fbs b/tensorflow/contrib/lite/schema/schema.fbs
index 4c339317cb..3da3188c3a 100644
--- a/tensorflow/contrib/lite/schema/schema.fbs
+++ b/tensorflow/contrib/lite/schema/schema.fbs
@@ -175,6 +175,7 @@ enum BuiltinOperator : byte {
REDUCE_ANY = 91,
SQUARE = 92,
ZEROS_LIKE = 93,
+ FILL = 94,
}
// Options for the builtin operators.
@@ -246,6 +247,7 @@ union BuiltinOptions {
FloorDivOptions,
SquareOptions,
ZerosLikeOptions,
+ FillOptions,
}
enum Padding : byte { SAME, VALID }
@@ -593,6 +595,9 @@ table SquareOptions {
table ZerosLikeOptions {
}
+table FillOptions {
+}
+
// An OperatorCode can be an enum value (BuiltinOperator) if the operator is a
// builtin, or a string if the operator is custom.
table OperatorCode {
diff --git a/tensorflow/contrib/lite/schema/schema_generated.h b/tensorflow/contrib/lite/schema/schema_generated.h
index 03c227f987..c7a59cabc5 100755
--- a/tensorflow/contrib/lite/schema/schema_generated.h
+++ b/tensorflow/contrib/lite/schema/schema_generated.h
@@ -232,6 +232,9 @@ struct SquareOptionsT;
struct ZerosLikeOptions;
struct ZerosLikeOptionsT;
+struct FillOptions;
+struct FillOptionsT;
+
struct OperatorCode;
struct OperatorCodeT;
@@ -391,11 +394,12 @@ enum BuiltinOperator {
BuiltinOperator_REDUCE_ANY = 91,
BuiltinOperator_SQUARE = 92,
BuiltinOperator_ZEROS_LIKE = 93,
+ BuiltinOperator_FILL = 94,
BuiltinOperator_MIN = BuiltinOperator_ADD,
- BuiltinOperator_MAX = BuiltinOperator_ZEROS_LIKE
+ BuiltinOperator_MAX = BuiltinOperator_FILL
};
-inline BuiltinOperator (&EnumValuesBuiltinOperator())[93] {
+inline BuiltinOperator (&EnumValuesBuiltinOperator())[94] {
static BuiltinOperator values[] = {
BuiltinOperator_ADD,
BuiltinOperator_AVERAGE_POOL_2D,
@@ -489,7 +493,8 @@ inline BuiltinOperator (&EnumValuesBuiltinOperator())[93] {
BuiltinOperator_FLOOR_DIV,
BuiltinOperator_REDUCE_ANY,
BuiltinOperator_SQUARE,
- BuiltinOperator_ZEROS_LIKE
+ BuiltinOperator_ZEROS_LIKE,
+ BuiltinOperator_FILL
};
return values;
}
@@ -590,6 +595,7 @@ inline const char **EnumNamesBuiltinOperator() {
"REDUCE_ANY",
"SQUARE",
"ZEROS_LIKE",
+ "FILL",
nullptr
};
return names;
@@ -669,11 +675,12 @@ enum BuiltinOptions {
BuiltinOptions_FloorDivOptions = 65,
BuiltinOptions_SquareOptions = 66,
BuiltinOptions_ZerosLikeOptions = 67,
+ BuiltinOptions_FillOptions = 68,
BuiltinOptions_MIN = BuiltinOptions_NONE,
- BuiltinOptions_MAX = BuiltinOptions_ZerosLikeOptions
+ BuiltinOptions_MAX = BuiltinOptions_FillOptions
};
-inline BuiltinOptions (&EnumValuesBuiltinOptions())[68] {
+inline BuiltinOptions (&EnumValuesBuiltinOptions())[69] {
static BuiltinOptions values[] = {
BuiltinOptions_NONE,
BuiltinOptions_Conv2DOptions,
@@ -742,7 +749,8 @@ inline BuiltinOptions (&EnumValuesBuiltinOptions())[68] {
BuiltinOptions_UnpackOptions,
BuiltinOptions_FloorDivOptions,
BuiltinOptions_SquareOptions,
- BuiltinOptions_ZerosLikeOptions
+ BuiltinOptions_ZerosLikeOptions,
+ BuiltinOptions_FillOptions
};
return values;
}
@@ -817,6 +825,7 @@ inline const char **EnumNamesBuiltinOptions() {
"FloorDivOptions",
"SquareOptions",
"ZerosLikeOptions",
+ "FillOptions",
nullptr
};
return names;
@@ -1099,6 +1108,10 @@ template<> struct BuiltinOptionsTraits<ZerosLikeOptions> {
static const BuiltinOptions enum_value = BuiltinOptions_ZerosLikeOptions;
};
+template<> struct BuiltinOptionsTraits<FillOptions> {
+ static const BuiltinOptions enum_value = BuiltinOptions_FillOptions;
+};
+
struct BuiltinOptionsUnion {
BuiltinOptions type;
void *value;
@@ -1666,6 +1679,14 @@ struct BuiltinOptionsUnion {
return type == BuiltinOptions_ZerosLikeOptions ?
reinterpret_cast<const ZerosLikeOptionsT *>(value) : nullptr;
}
+ FillOptionsT *AsFillOptions() {
+ return type == BuiltinOptions_FillOptions ?
+ reinterpret_cast<FillOptionsT *>(value) : nullptr;
+ }
+ const FillOptionsT *AsFillOptions() const {
+ return type == BuiltinOptions_FillOptions ?
+ reinterpret_cast<const FillOptionsT *>(value) : nullptr;
+ }
};
bool VerifyBuiltinOptions(flatbuffers::Verifier &verifier, const void *obj, BuiltinOptions type);
@@ -5949,6 +5970,46 @@ inline flatbuffers::Offset<ZerosLikeOptions> CreateZerosLikeOptions(
flatbuffers::Offset<ZerosLikeOptions> CreateZerosLikeOptions(flatbuffers::FlatBufferBuilder &_fbb, const ZerosLikeOptionsT *_o, const flatbuffers::rehasher_function_t *_rehasher = nullptr);
+struct FillOptionsT : public flatbuffers::NativeTable {
+ typedef FillOptions TableType;
+ FillOptionsT() {
+ }
+};
+
+struct FillOptions FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table {
+ typedef FillOptionsT NativeTableType;
+ bool Verify(flatbuffers::Verifier &verifier) const {
+ return VerifyTableStart(verifier) &&
+ verifier.EndTable();
+ }
+ FillOptionsT *UnPack(const flatbuffers::resolver_function_t *_resolver = nullptr) const;
+ void UnPackTo(FillOptionsT *_o, const flatbuffers::resolver_function_t *_resolver = nullptr) const;
+ static flatbuffers::Offset<FillOptions> Pack(flatbuffers::FlatBufferBuilder &_fbb, const FillOptionsT* _o, const flatbuffers::rehasher_function_t *_rehasher = nullptr);
+};
+
+struct FillOptionsBuilder {
+ flatbuffers::FlatBufferBuilder &fbb_;
+ flatbuffers::uoffset_t start_;
+ explicit FillOptionsBuilder(flatbuffers::FlatBufferBuilder &_fbb)
+ : fbb_(_fbb) {
+ start_ = fbb_.StartTable();
+ }
+ FillOptionsBuilder &operator=(const FillOptionsBuilder &);
+ flatbuffers::Offset<FillOptions> Finish() {
+ const auto end = fbb_.EndTable(start_);
+ auto o = flatbuffers::Offset<FillOptions>(end);
+ return o;
+ }
+};
+
+inline flatbuffers::Offset<FillOptions> CreateFillOptions(
+ flatbuffers::FlatBufferBuilder &_fbb) {
+ FillOptionsBuilder builder_(_fbb);
+ return builder_.Finish();
+}
+
+flatbuffers::Offset<FillOptions> CreateFillOptions(flatbuffers::FlatBufferBuilder &_fbb, const FillOptionsT *_o, const flatbuffers::rehasher_function_t *_rehasher = nullptr);
+
struct OperatorCodeT : public flatbuffers::NativeTable {
typedef OperatorCode TableType;
BuiltinOperator builtin_code;
@@ -6283,6 +6344,9 @@ struct Operator FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table {
const ZerosLikeOptions *builtin_options_as_ZerosLikeOptions() const {
return builtin_options_type() == BuiltinOptions_ZerosLikeOptions ? static_cast<const ZerosLikeOptions *>(builtin_options()) : nullptr;
}
+ const FillOptions *builtin_options_as_FillOptions() const {
+ return builtin_options_type() == BuiltinOptions_FillOptions ? static_cast<const FillOptions *>(builtin_options()) : nullptr;
+ }
const flatbuffers::Vector<uint8_t> *custom_options() const {
return GetPointer<const flatbuffers::Vector<uint8_t> *>(VT_CUSTOM_OPTIONS);
}
@@ -6582,6 +6646,10 @@ template<> inline const ZerosLikeOptions *Operator::builtin_options_as<ZerosLike
return builtin_options_as_ZerosLikeOptions();
}
+template<> inline const FillOptions *Operator::builtin_options_as<FillOptions>() const {
+ return builtin_options_as_FillOptions();
+}
+
struct OperatorBuilder {
flatbuffers::FlatBufferBuilder &fbb_;
flatbuffers::uoffset_t start_;
@@ -8873,6 +8941,29 @@ inline flatbuffers::Offset<ZerosLikeOptions> CreateZerosLikeOptions(flatbuffers:
_fbb);
}
+inline FillOptionsT *FillOptions::UnPack(const flatbuffers::resolver_function_t *_resolver) const {
+ auto _o = new FillOptionsT();
+ UnPackTo(_o, _resolver);
+ return _o;
+}
+
+inline void FillOptions::UnPackTo(FillOptionsT *_o, const flatbuffers::resolver_function_t *_resolver) const {
+ (void)_o;
+ (void)_resolver;
+}
+
+inline flatbuffers::Offset<FillOptions> FillOptions::Pack(flatbuffers::FlatBufferBuilder &_fbb, const FillOptionsT* _o, const flatbuffers::rehasher_function_t *_rehasher) {
+ return CreateFillOptions(_fbb, _o, _rehasher);
+}
+
+inline flatbuffers::Offset<FillOptions> CreateFillOptions(flatbuffers::FlatBufferBuilder &_fbb, const FillOptionsT *_o, const flatbuffers::rehasher_function_t *_rehasher) {
+ (void)_rehasher;
+ (void)_o;
+ struct _VectorArgs { flatbuffers::FlatBufferBuilder *__fbb; const FillOptionsT* __o; const flatbuffers::rehasher_function_t *__rehasher; } _va = { &_fbb, _o, _rehasher}; (void)_va;
+ return tflite::CreateFillOptions(
+ _fbb);
+}
+
inline OperatorCodeT *OperatorCode::UnPack(const flatbuffers::resolver_function_t *_resolver) const {
auto _o = new OperatorCodeT();
UnPackTo(_o, _resolver);
@@ -9330,6 +9421,10 @@ inline bool VerifyBuiltinOptions(flatbuffers::Verifier &verifier, const void *ob
auto ptr = reinterpret_cast<const ZerosLikeOptions *>(obj);
return verifier.VerifyTable(ptr);
}
+ case BuiltinOptions_FillOptions: {
+ auto ptr = reinterpret_cast<const FillOptions *>(obj);
+ return verifier.VerifyTable(ptr);
+ }
default: return false;
}
}
@@ -9616,6 +9711,10 @@ inline void *BuiltinOptionsUnion::UnPack(const void *obj, BuiltinOptions type, c
auto ptr = reinterpret_cast<const ZerosLikeOptions *>(obj);
return ptr->UnPack(resolver);
}
+ case BuiltinOptions_FillOptions: {
+ auto ptr = reinterpret_cast<const FillOptions *>(obj);
+ return ptr->UnPack(resolver);
+ }
default: return nullptr;
}
}
@@ -9890,6 +9989,10 @@ inline flatbuffers::Offset<void> BuiltinOptionsUnion::Pack(flatbuffers::FlatBuff
auto ptr = reinterpret_cast<const ZerosLikeOptionsT *>(value);
return CreateZerosLikeOptions(_fbb, ptr, _rehasher).Union();
}
+ case BuiltinOptions_FillOptions: {
+ auto ptr = reinterpret_cast<const FillOptionsT *>(value);
+ return CreateFillOptions(_fbb, ptr, _rehasher).Union();
+ }
default: return 0;
}
}
@@ -10164,6 +10267,10 @@ inline BuiltinOptionsUnion::BuiltinOptionsUnion(const BuiltinOptionsUnion &u) FL
value = new ZerosLikeOptionsT(*reinterpret_cast<ZerosLikeOptionsT *>(u.value));
break;
}
+ case BuiltinOptions_FillOptions: {
+ value = new FillOptionsT(*reinterpret_cast<FillOptionsT *>(u.value));
+ break;
+ }
default:
break;
}
@@ -10506,6 +10613,11 @@ inline void BuiltinOptionsUnion::Reset() {
delete ptr;
break;
}
+ case BuiltinOptions_FillOptions: {
+ auto ptr = reinterpret_cast<FillOptionsT *>(value);
+ delete ptr;
+ break;
+ }
default: break;
}
value = nullptr;