aboutsummaryrefslogtreecommitdiff
path: root/python/tosa
diff options
context:
space:
mode:
Diffstat (limited to 'python/tosa')
-rw-r--r--python/tosa/ConvAttribute.py15
-rw-r--r--python/tosa/TransposeConvAttribute.py15
2 files changed, 28 insertions, 2 deletions
diff --git a/python/tosa/ConvAttribute.py b/python/tosa/ConvAttribute.py
index b35b67c..dfa75dc 100644
--- a/python/tosa/ConvAttribute.py
+++ b/python/tosa/ConvAttribute.py
@@ -130,8 +130,15 @@ class ConvAttribute(object):
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
return False
+ # ConvAttribute
+ def AccType(self):
+ o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
+ if o != 0:
+ return self._tab.Get(flatbuffers.number_types.Uint32Flags, o + self._tab.Pos)
+ return 0
+
def ConvAttributeStart(builder):
- builder.StartObject(6)
+ builder.StartObject(7)
def Start(builder):
ConvAttributeStart(builder)
@@ -190,6 +197,12 @@ def ConvAttributeAddLocalBound(builder, localBound):
def AddLocalBound(builder, localBound):
ConvAttributeAddLocalBound(builder, localBound)
+def ConvAttributeAddAccType(builder, accType):
+ builder.PrependUint32Slot(6, accType, 0)
+
+def AddAccType(builder, accType):
+ ConvAttributeAddAccType(builder, accType)
+
def ConvAttributeEnd(builder):
return builder.EndObject()
diff --git a/python/tosa/TransposeConvAttribute.py b/python/tosa/TransposeConvAttribute.py
index a74a433..e5397a8 100644
--- a/python/tosa/TransposeConvAttribute.py
+++ b/python/tosa/TransposeConvAttribute.py
@@ -130,8 +130,15 @@ class TransposeConvAttribute(object):
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
return False
+ # TransposeConvAttribute
+ def AccType(self):
+ o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
+ if o != 0:
+ return self._tab.Get(flatbuffers.number_types.Uint32Flags, o + self._tab.Pos)
+ return 0
+
def TransposeConvAttributeStart(builder):
- builder.StartObject(6)
+ builder.StartObject(7)
def Start(builder):
TransposeConvAttributeStart(builder)
@@ -190,6 +197,12 @@ def TransposeConvAttributeAddLocalBound(builder, localBound):
def AddLocalBound(builder, localBound):
TransposeConvAttributeAddLocalBound(builder, localBound)
+def TransposeConvAttributeAddAccType(builder, accType):
+ builder.PrependUint32Slot(6, accType, 0)
+
+def AddAccType(builder, accType):
+ TransposeConvAttributeAddAccType(builder, accType)
+
def TransposeConvAttributeEnd(builder):
return builder.EndObject()