From e1072a9ed871fd474e7b09b7a74ae7be5f0a6f78 Mon Sep 17 00:00:00 2001 From: Jeremy Johnson Date: Tue, 27 Sep 2022 12:44:11 +0100 Subject: Rename FLOAT type to FP32 Signed-off-by: Jeremy Johnson Change-Id: I2cd6ae85f59fd62a62b7f03ae9667228b54e82db --- python/serializer/tosa_serializer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'python/serializer/tosa_serializer.py') diff --git a/python/serializer/tosa_serializer.py b/python/serializer/tosa_serializer.py index fb89563..f4e146c 100644 --- a/python/serializer/tosa_serializer.py +++ b/python/serializer/tosa_serializer.py @@ -56,7 +56,7 @@ DTypeNames = [ "INT16", "INT32", "INT48", - "FLOAT", + "FP32", "UINT16", "FP16", ] @@ -378,7 +378,7 @@ class TosaSerializerTensor: self.shape = shape self.dtype = dtype - if dtype == DType.FLOAT: + if dtype == DType.FP32: fntype = np.float32 elif dtype == DType.FP16: fntype = np.float16 @@ -466,7 +466,7 @@ class TosaSerializerTensor: elif self.dtype == DType.FP16: np_arr = np.array(self.data, dtype=np.float16) u8_data.extend(np_arr.view(np.uint8)) - elif self.dtype == DType.FLOAT: + elif self.dtype == DType.FP32: for val in self.data: b = struct.pack("!f", val) u8_data.extend([b[3], b[2], b[1], b[0]]) -- cgit v1.2.1