diff options
| author | Rickard Bolin <rickard.bolin@arm.com> | 2023-04-24 10:39:41 +0000 |
|---|---|---|
| committer | Rickard Bolin <rickard.bolin@arm.com> | 2023-04-25 11:32:40 +0000 |
| commit | 3fd5d839a29e8f0a00638e1c6f86695a5d30d41a (patch) | |
| tree | 16fb2f818073b09b16c8ec610dc072b856f1fdd0 /ethosu/vela/tflite | |
| parent | c4268bf407048c7899c8501dd1223a777f8c4963 (diff) | |
| download | ethos-u-vela-3fd5d839a29e8f0a00638e1c6f86695a5d30d41a.tar.gz | |
MLBEDSW-6954: Update to TensorFlow 2.11
Updated FlatBuffers autogenerated files to TensorFlow 2.11
Change-Id: Ia39d30b06e9a37c9ab119d501ebf442f32167afe
Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
Diffstat (limited to 'ethosu/vela/tflite')
| -rw-r--r-- | ethosu/vela/tflite/TensorType.py | 1 | ||||
| -rw-r--r-- | ethosu/vela/tflite/TransposeConvOptions.py | 12 |
2 files changed, 12 insertions, 1 deletions
diff --git a/ethosu/vela/tflite/TensorType.py b/ethosu/vela/tflite/TensorType.py index 621d0c54..604d4362 100644 --- a/ethosu/vela/tflite/TensorType.py +++ b/ethosu/vela/tflite/TensorType.py @@ -20,3 +20,4 @@ class TensorType(object): VARIANT = 14 UINT32 = 15 UINT16 = 16 + INT4 = 17 diff --git a/ethosu/vela/tflite/TransposeConvOptions.py b/ethosu/vela/tflite/TransposeConvOptions.py index a11a79e9..af9159c2 100644 --- a/ethosu/vela/tflite/TransposeConvOptions.py +++ b/ethosu/vela/tflite/TransposeConvOptions.py @@ -49,7 +49,14 @@ class TransposeConvOptions(object): return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos) return 0 -def TransposeConvOptionsStart(builder): builder.StartObject(3) + # TransposeConvOptions + def FusedActivationFunction(self): + o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10)) + if o != 0: + return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos) + return 0 + +def TransposeConvOptionsStart(builder): builder.StartObject(4) def Start(builder): return TransposeConvOptionsStart(builder) def TransposeConvOptionsAddPadding(builder, padding): builder.PrependInt8Slot(0, padding, 0) @@ -61,6 +68,9 @@ def AddStrideW(builder, strideW): def TransposeConvOptionsAddStrideH(builder, strideH): builder.PrependInt32Slot(2, strideH, 0) def AddStrideH(builder, strideH): return TransposeConvOptionsAddStrideH(builder, strideH) +def TransposeConvOptionsAddFusedActivationFunction(builder, fusedActivationFunction): builder.PrependInt8Slot(3, fusedActivationFunction, 0) +def AddFusedActivationFunction(builder, fusedActivationFunction): + return TransposeConvOptionsAddFusedActivationFunction(builder, fusedActivationFunction) def TransposeConvOptionsEnd(builder): return builder.EndObject() def End(builder): return TransposeConvOptionsEnd(builder)
\ No newline at end of file |