diff options
Diffstat (limited to 'ethosu/vela/npu_performance.py')
-rw-r--r-- | ethosu/vela/npu_performance.py | 24 |
1 files changed, 18 insertions, 6 deletions
diff --git a/ethosu/vela/npu_performance.py b/ethosu/vela/npu_performance.py index 4ffca496..0c8a9073 100644 --- a/ethosu/vela/npu_performance.py +++ b/ethosu/vela/npu_performance.py @@ -59,14 +59,26 @@ class PassCycles(IntEnum): Size = auto() def display_name(self): - return ("NPU", "SRAM Access", "DRAM Access", "On-chip Flash Access", "Off-chip Flash Access", "Total", "Size",)[ - self.value - ] + return ( + "NPU", + "SRAM Access", + "DRAM Access", + "On-chip Flash Access", + "Off-chip Flash Access", + "Total", + "Size", + )[self.value] def identifier_name(self): - return ("npu", "sram_access", "dram_access", "on_chip_flash_access", "off_chip_flash_access", "total", "size",)[ - self.value - ] + return ( + "npu", + "sram_access", + "dram_access", + "on_chip_flash_access", + "off_chip_flash_access", + "total", + "size", + )[self.value] @staticmethod def all(): |