diff options
author | Kevin Cheng <kevin.cheng@arm.com> | 2021-08-31 16:14:03 -0700 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2021-09-15 23:58:06 +0000 |
commit | 93a1628bc3dd48d9ba099de503b586a561b4751f (patch) | |
tree | adab77805c3d78cf3b30b00684e8a76316e11477 /verif/tosa_serializer.py | |
parent | e3d6a8ffe0fffaf9d29167b03509b85a2f4d8308 (diff) | |
download | reference_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.py | 44 |
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 |