aboutsummaryrefslogtreecommitdiff
path: root/python/tosa/FullyConnectedAttribute.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/tosa/FullyConnectedAttribute.py')
-rw-r--r--python/tosa/FullyConnectedAttribute.py12
1 files changed, 11 insertions, 1 deletions
diff --git a/python/tosa/FullyConnectedAttribute.py b/python/tosa/FullyConnectedAttribute.py
index 892b0da..546ec60 100644
--- a/python/tosa/FullyConnectedAttribute.py
+++ b/python/tosa/FullyConnectedAttribute.py
@@ -42,7 +42,14 @@ class FullyConnectedAttribute(object):
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
return 0
-def FullyConnectedAttributeStart(builder): builder.StartObject(2)
+ # FullyConnectedAttribute
+ def AccumDtype(self):
+ o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
+ if o != 0:
+ return self._tab.Get(flatbuffers.number_types.Uint32Flags, o + self._tab.Pos)
+ return 0
+
+def FullyConnectedAttributeStart(builder): builder.StartObject(3)
def Start(builder):
return FullyConnectedAttributeStart(builder)
def FullyConnectedAttributeAddInputZp(builder, inputZp): builder.PrependInt32Slot(0, inputZp, 0)
@@ -51,6 +58,9 @@ def AddInputZp(builder, inputZp):
def FullyConnectedAttributeAddWeightZp(builder, weightZp): builder.PrependInt32Slot(1, weightZp, 0)
def AddWeightZp(builder, weightZp):
return FullyConnectedAttributeAddWeightZp(builder, weightZp)
+def FullyConnectedAttributeAddAccumDtype(builder, accumDtype): builder.PrependUint32Slot(2, accumDtype, 0)
+def AddAccumDtype(builder, accumDtype):
+ return FullyConnectedAttributeAddAccumDtype(builder, accumDtype)
def FullyConnectedAttributeEnd(builder): return builder.EndObject()
def End(builder):
return FullyConnectedAttributeEnd(builder) \ No newline at end of file