MLBEDSW-2836 Change sets to tuples
Replace conditional checks against sets with tuples.
If not requiring uniqueness, or complex set operations, it is quicker to
use tuples instead.
Signed-off-by: Michael McGeagh <michael.mcgeagh@arm.com>
Change-Id: Ie8732c8d46067244963936c53f0ec81adda50372
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py
index b07b4dc..f6e628c 100644
--- a/ethosu/vela/tensor.py
+++ b/ethosu/vela/tensor.py
@@ -557,8 +557,10 @@
return self.consumer_list
def get_address_ranges_for_coordinates(self, start_coord, end_coord):
- if self.sub_purpose in set(
- (TensorSubPurpose.RollingBufferX, TensorSubPurpose.RollingBufferY, TensorSubPurpose.RollingBufferXY)
+ if self.sub_purpose in (
+ TensorSubPurpose.RollingBufferX,
+ TensorSubPurpose.RollingBufferY,
+ TensorSubPurpose.RollingBufferXY,
):
# build dummy coordinates that cover the entire buffer
start_coord = [0] * len(start_coord)
@@ -637,7 +639,7 @@
augmented_shape[1] = 1
else:
- assert self.format in set((TensorFormat.Unknown, TensorFormat.WeightsCompressed))
+ assert self.format in (TensorFormat.Unknown, TensorFormat.WeightsCompressed)
return None, None
strides = [0] * len(augmented_shape)
@@ -774,9 +776,7 @@
return address_offset
def is_allocated_in_tensor_arena(self, scratch_tensor_mem_area):
- if self.mem_area == scratch_tensor_mem_area and (self.mem_type in set((MemType.Scratch, MemType.Scratch_fast))):
- return True
- return False
+ return (self.mem_area == scratch_tensor_mem_area) and (self.mem_type in (MemType.Scratch, MemType.Scratch_fast))
def equivalent(self, tens):
return self.equivalence_id == tens.equivalence_id