diff options
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): |