Update script

llvm/projects/hpvm-tensor-rt/bin/time_jetson_profiles.py

@@ -9,6 +9,7 @@ class Benchmark:
     self.binary_path = ""
     self.binary_time = 0
     self.num_layers = 0
+    self.data_size = 0
 
 
 
@@ -18,6 +19,8 @@ ResNet50.binary_path = "resnet_imagenet"
 ResNet50.binary_time = 3.85 * 100  # 50 images * 100 batches
 ResNet50.batch_time = 3.85  # Time for batch with 50 images
 ResNet50.num_layers = 53
+ResNet50.data_size = 50 * 3 * 224 * 224
+
 
 
 VGG16_ImageNet = Benchmark()
@@ -26,6 +29,8 @@ VGG16_ImageNet.binary_path = "vgg16_imagenet"
 VGG16_ImageNet.binary_time = 4.55 * 100  # 50 images * 100 batches
 VGG16_ImageNet.batch_time = 4.55
 VGG16_ImageNet.num_layers = 16
+VGG16_ImageNet.data_size = 50 * 3 * 224 * 224
+
 
 
 AlexNet_ImageNet = Benchmark()
@@ -33,6 +38,7 @@ AlexNet_ImageNet.binary_path = "alexnet_imagenet"
 AlexNet_ImageNet.binary_time = 0.66 * 100  
 AlexNet_ImageNet.batch_time = 0.66
 AlexNet_ImageNet.num_layers = 8
+AlexNet_ImageNet.data_size = 50 * 3 * 224 * 224