Update benchmarks.yml

Author: juliagmt-google

.github/workflows/benchmarks.yml

@@ -78,10 +78,10 @@ jobs:
         run: |
           cat << EOF > isolated_convolution.hlo
           HloModule convolution.167:
-          ENTRY %convolution.167 (parameter.0: f32[16,28,28,128], parameter.1: f32[3,3,128,128]) -> f32[16,28,28,128] {
-            %parameter.0 = f32[16,28,28,128]{3,0,2,1} parameter(0)
-            %parameter.1 = f32[3,3,128,128]{3,2,1,0} parameter(1)
-            ROOT %convolution.167 = f32[16,28,28,128]{3,0,2,1} convolution(f32[16,28,28,128]{3,0,2,1} %parameter.0, f32[3,3,128,128]{3,2,1,0} %parameter.1), window={size=3x3 pad=1_1x1_1}, dim_labels=b01f_01oi->b01f
+          ENTRY %convolution.167 (parameter.0: f32[64,28,28,128]{2,1,0,3}, parameter.1: f32[3,3,128,128]{2,1,0,3}) -> f32[64,28,28,128]{2,1,0,3} {
+            %parameter.0 = f32[64,28,28,128]{2,1,0,3} parameter(0)
+            %parameter.1 = f32[3,3,128,128]{2,1,0,3} parameter(1)
+            ROOT %convolution.167 = f32[64,28,28,128]{2,1,0,3} convolution(f32[64,28,28,128]{2,1,0,3} %parameter.0, f32[3,3,128,128]{2,1,0,3} %parameter.1), window={size=3x3 pad=1_1x1_1}, dim_labels=bf01_01oi->bf01
           }
           EOF