diff options
author | Tai Ly <tai.ly@arm.com> | 2024-04-24 16:03:22 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2024-04-26 21:26:16 +0000 |
commit | 3aebe2bd863d6e0cb82171984cd49e5ad516d0db (patch) | |
tree | e56471af9df1534d868e8c65353accb72857047d /python/tosa/PoolAttribute.py | |
parent | 36ced1df313cf80edc91efe41facb1ab3a81b220 (diff) | |
download | serialization_lib-main.tar.gz |
Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: I9ab0e8ab4e3c37f1dd135b57661b25e18a5df5cb
Diffstat (limited to 'python/tosa/PoolAttribute.py')
-rw-r--r-- | python/tosa/PoolAttribute.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/python/tosa/PoolAttribute.py b/python/tosa/PoolAttribute.py index c13e038..831d43b 100644 --- a/python/tosa/PoolAttribute.py +++ b/python/tosa/PoolAttribute.py @@ -145,7 +145,7 @@ def AddPad(builder, pad): def PoolAttributeStartPadVector(builder, numElems): return builder.StartVector(4, numElems, 4) -def StartPadVector(builder, numElems: int) -> int: +def StartPadVector(builder, numElems): return PoolAttributeStartPadVector(builder, numElems) def PoolAttributeAddKernel(builder, kernel): @@ -157,7 +157,7 @@ def AddKernel(builder, kernel): def PoolAttributeStartKernelVector(builder, numElems): return builder.StartVector(4, numElems, 4) -def StartKernelVector(builder, numElems: int) -> int: +def StartKernelVector(builder, numElems): return PoolAttributeStartKernelVector(builder, numElems) def PoolAttributeAddStride(builder, stride): @@ -169,7 +169,7 @@ def AddStride(builder, stride): def PoolAttributeStartStrideVector(builder, numElems): return builder.StartVector(4, numElems, 4) -def StartStrideVector(builder, numElems: int) -> int: +def StartStrideVector(builder, numElems): return PoolAttributeStartStrideVector(builder, numElems) def PoolAttributeAddInputZp(builder, inputZp): |