To automate the utilization measurements, we identified computational units in all 35 image classification architectures supported by Torchvision [1]}. The classification architectures of AI models are extracted into trace-based tree or block class-based tree textual representations - see Figure REF  (left and middle). Using the TorchScript vision library [1]}, these tree representations can be converted into a DiGraph representation [3]} (see Figure REF  right) that can be pseudo-colored and visualized according to the block diagram shown in Figure REF .
