aboutsummaryrefslogtreecommitdiff
path: root/verif/tosa_serializer.py
diff options
context:
space:
mode:
authorKevin Cheng <kevin.cheng@arm.com>2021-08-31 16:14:03 -0700
committerEric Kunze <eric.kunze@arm.com>2021-09-15 23:58:06 +0000
commit93a1628bc3dd48d9ba099de503b586a561b4751f (patch)
treeadab77805c3d78cf3b30b00684e8a76316e11477 /verif/tosa_serializer.py
parente3d6a8ffe0fffaf9d29167b03509b85a2f4d8308 (diff)
downloadreference_model-93a1628bc3dd48d9ba099de503b586a561b4751f.tar.gz
Rename attribute: Pool2d, Conv2d, TransposeConv2d -> Pool, Conv, TransposeConv
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com> Change-Id: I466dd1dcf5230e8e07df202ba88515e775e04a1e
Diffstat (limited to 'verif/tosa_serializer.py')
-rw-r--r--verif/tosa_serializer.py44
1 files changed, 22 insertions, 22 deletions
diff --git a/verif/tosa_serializer.py b/verif/tosa_serializer.py
index e9ec027..22ebb52 100644
--- a/verif/tosa_serializer.py
+++ b/verif/tosa_serializer.py
@@ -135,36 +135,36 @@ class TosaSerializerAttribute(TosaSerializerUnion):
def __init__(self):
super().__init__()
- def Pool2dAttribute(self, kernel, stride, padding):
- from tosa import Pool2dAttribute as a, Attribute
+ def PoolAttribute(self, kernel, stride, padding):
+ from tosa import PoolAttribute as a, Attribute
- self.utype = Attribute.Attribute().Pool2dAttribute
+ self.utype = Attribute.Attribute().PoolAttribute
- self.optFcns = (a.Pool2dAttributeStart, a.Pool2dAttributeEnd)
- self.intvecs.append((a.Pool2dAttributeAddPadding, padding))
- self.intvecs.append((a.Pool2dAttributeAddKernel, kernel))
- self.intvecs.append((a.Pool2dAttributeAddStride, stride))
+ self.optFcns = (a.PoolAttributeStart, a.PoolAttributeEnd)
+ self.intvecs.append((a.PoolAttributeAddPadding, padding))
+ self.intvecs.append((a.PoolAttributeAddKernel, kernel))
+ self.intvecs.append((a.PoolAttributeAddStride, stride))
- def Conv2dAttribute(self, padding, stride, dilation):
- from tosa import Conv2dAttribute as a, Attribute
+ def ConvAttribute(self, padding, stride, dilation):
+ from tosa import ConvAttribute as a, Attribute
- self.utype = Attribute.Attribute().Conv2dAttribute
- self.optFcns = (a.Conv2dAttributeStart, a.Conv2dAttributeEnd)
+ self.utype = Attribute.Attribute().ConvAttribute
+ self.optFcns = (a.ConvAttributeStart, a.ConvAttributeEnd)
- self.intvecs.append((a.Conv2dAttributeAddPadding, padding))
- self.intvecs.append((a.Conv2dAttributeAddStride, stride))
- self.intvecs.append((a.Conv2dAttributeAddDilation, dilation))
+ self.intvecs.append((a.ConvAttributeAddPadding, padding))
+ self.intvecs.append((a.ConvAttributeAddStride, stride))
+ self.intvecs.append((a.ConvAttributeAddDilation, dilation))
- def TransposeConv2DAttribute(self, outpad, stride, dilation, output_shape):
- from tosa import TransposeConv2dAttribute as a, Attribute
+ def TransposeConvAttribute(self, outpad, stride, dilation, output_shape):
+ from tosa import TransposeConvAttribute as a, Attribute
- self.utype = Attribute.Attribute().TransposeConv2dAttribute
- self.optFcns = (a.TransposeConv2dAttributeStart, a.TransposeConv2dAttributeEnd)
+ self.utype = Attribute.Attribute().TransposeConvAttribute
+ self.optFcns = (a.TransposeConvAttributeStart, a.TransposeConvAttributeEnd)
- self.intvecs.append((a.TransposeConv2dAttributeAddOutpad, outpad))
- self.intvecs.append((a.TransposeConv2dAttributeAddStride, stride))
- self.intvecs.append((a.TransposeConv2dAttributeAddDilation, dilation))
- self.intvecs.append((a.TransposeConv2dAttributeAddOutputShape, output_shape))
+ self.intvecs.append((a.TransposeConvAttributeAddOutpad, outpad))
+ self.intvecs.append((a.TransposeConvAttributeAddStride, stride))
+ self.intvecs.append((a.TransposeConvAttributeAddDilation, dilation))
+ self.intvecs.append((a.TransposeConvAttributeAddOutputShape, output_shape))
def ReluNAttribute(self, maxint, maxfp):
from tosa import ReluNAttribute as a, Attribute