Increase shard count of //third_party/tensorflow/python:nn_batchnorm_test to avoid timeouts
Increase shard count of //third_party/tensorflow/python:nn_batchnorm_test to avoid timeouts

When run under asan, the test runs for about 5 minutes, and sometimes
longer, causing frequent timeouts.

This change increases the shard count of the test to 4, which brings the run time
of the longest running shard under asan to about 2 minutes.

PiperOrigin-RevId: 183414888

adit-chandra/tensorflow,tensorflow/tensorflow,xzturn/tensorflow,kobejean/tensorflow,Xeralux/tensorflow,tensorflow/tensorflow-pywrap_saved_model,aam-at/tensorflow,girving/tensorflow,davidzchen/tensorflow,benoitsteiner/tensorflow-xsmm,ravindrapanda/tensorflow,drpngx/tensorflow,ravindrapanda/tensorflow,dendisuhubdy/tensorflow,karllessard/tensorflow,xodus7/tensorflow,gunan/tensorflow,davidzchen/tensorflow,Bismarrck/tensorflow,annarev/tensorflow,kevin-coder/tensorflow-fork,gautam1858/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,jhseu/tensorflow,Xeralux/tensorflow,dongjoon-hyun/tensorflow,ageron/tensorflow,asimshankar/tensorflow,nburn42/tensorflow,aldian/tensorflow,jbedorf/tensorflow,tensorflow/tensorflow,asimshankar/tensorflow,nolanliou/tensorflow,adit-chandra/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,jbedorf/tensorflow,seanli9jan/tensorflow,yongtang/tensorflow,alsrgv/tensorflow,jendap/tensorflow,adit-chandra/tensorflow,davidzchen/tensorflow,gautam1858/tensorflow,freedomtan/tensorflow,jendap/tensorflow,girving/tensorflow,paolodedios/tensorflow,ZhangXinNan/tensorflow,yanchen036/tensorflow,adit-chandra/tensorflow,apark263/tensorflow,brchiu/tensorflow,xodus7/tensorflow,eaplatanios/tensorflow,arborh/tensorflow,seanli9jan/tensorflow,adit-chandra/tensorflow,ghchinoy/tensorflow,Intel-tensorflow/tensorflow,apark263/tensorflow,eaplatanios/tensorflow,DavidNorman/tensorflow,seanli9jan/tensorflow,xzturn/tensorflow,alshedivat/tensorflow,xodus7/tensorflow,asimshankar/tensorflow,manipopopo/tensorflow,cxxgtxy/tensorflow,brchiu/tensorflow,karllessard/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,gojira/tensorflow,eaplatanios/tensorflow,Intel-Corporation/tensorflow,drpngx/tensorflow,ZhangXinNan/tensorflow,alshedivat/tensorflow,kevin-coder/tensorflow-fork,tensorflow/tensorflow-experimental_link_static_libraries_once,AnishShah/tensorflow,aam-at/tensorflow,theflofly/tensorflow,drpngx/tensorflow,freedomtan/tensorflow,brchiu/tensorflow,chemelnucfin/tensorflow,sarvex/tensorflow,av8ramit/tensorflow,hfp/tensorflow-xsmm,paolodedios/tensorflow,gunan/tensorflow,tensorflow/tensorflow-pywrap_saved_model,ppwwyyxx/tensorflow,renyi533/tensorflow,chemelnucfin/tensorflow,sarvex/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,tensorflow/tensorflow-pywrap_tf_optimizer,nolanliou/tensorflow,Bismarrck/tensorflow,manipopopo/tensorflow,seanli9jan/tensorflow,ppwwyyxx/tensorflow,dendisuhubdy/tensorflow,ghchinoy/tensorflow,Intel-tensorflow/tensorflow,yongtang/tensorflow,frreiss/tensorflow-fred,meteorcloudy/tensorflow,ppwwyyxx/tensorflow,Xeralux/tensorflow,arborh/tensorflow,gautam1858/tensorflow,DavidNorman/tensorflow,chemelnucfin/tensorflow,paolodedios/tensorflow,gautam1858/tensorflow,benoitsteiner/tensorflow-xsmm,xodus7/tensorflow,aldian/tensorflow,renyi533/tensorflow,gunan/tensorflow,davidzchen/tensorflow,snnn/tensorflow,jhseu/tensorflow,jalexvig/tensorflow,Bismarrck/tensorflow,zasdfgbnm/tensorflow,alsrgv/tensorflow,benoitsteiner/tensorflow-xsmm,alshedivat/tensorflow,asimshankar/tensorflow,manipopopo/tensorflow,dongjoon-hyun/tensorflow,xodus7/tensorflow,caisq/tensorflow,paolodedios/tensorflow,apark263/tensorflow,jhseu/tensorflow,frreiss/tensorflow-fred,yongtang/tensorflow,tensorflow/tensorflow,annarev/tensorflow,dancingdan/tensorflow,zasdfgbnm/tensorflow,aam-at/tensorflow,brchiu/tensorflow,dancingdan/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,nburn42/tensorflow,adit-chandra/tensorflow,renyi533/tensorflow,av8ramit/tensorflow,kobejean/tensorflow,AnishShah/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,adit-chandra/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,dendisuhubdy/tensorflow,petewarden/tensorflow,Intel-tensorflow/tensorflow,xzturn/tensorflow,Xeralux/tensorflow,Xeralux/tensorflow,jendap/tensorflow,chemelnucfin/tensorflow,girving/tensorflow,lukeiwanski/tensorflow,xzturn/tensorflow,karllessard/tensorflow,yanchen036/tensorflow,cxxgtxy/tensorflow,petewarden/tensorflow,dendisuhubdy/tensorflow,nolanliou/tensorflow,ZhangXinNan/tensorflow,jendap/tensorflow,manipopopo/tensorflow,allenlavoie/tensorflow,cxxgtxy/tensorflow,xodus7/tensorflow,alsrgv/tensorflow,jart/tensorflow,kobejean/tensorflow,dendisuhubdy/tensorflow,aam-at/tensorflow,ghchinoy/tensorflow,benoitsteiner/tensorflow-xsmm,alshedivat/tensorflow,jbedorf/tensorflow,freedomtan/tensorflow,lukeiwanski/tensorflow,kobejean/tensorflow,davidzchen/tensorflow,chemelnucfin/tensorflow,freedomtan/tensorflow,brchiu/tensorflow,hfp/tensorflow-xsmm,ghchinoy/tensorflow,rabipanda/tensorflow,yanchen036/tensorflow,eaplatanios/tensorflow,jart/tensorflow,paolodedios/tensorflow,ageron/tensorflow,ravindrapanda/tensorflow,kevin-coder/tensorflow-fork,xodus7/tensorflow,gautam1858/tensorflow,gunan/tensorflow,adit-chandra/tensorflow,gojira/tensorflow,seanli9jan/tensorflow,renyi533/tensorflow,tensorflow/tensorflow,lukeiwanski/tensorflow,renyi533/tensorflow,karllessard/tensorflow,frreiss/tensorflow-fred,ZhangXinNan/tensorflow,yongtang/tensorflow,hfp/tensorflow-xsmm,sarvex/tensorflow,snnn/tensorflow,Intel-Corporation/tensorflow,av8ramit/tensorflow,dendisuhubdy/tensorflow,ravindrapanda/tensorflow,kobejean/tensorflow,kobejean/tensorflow,caisq/tensorflow,jhseu/tensorflow,petewarden/tensorflow,jart/tensorflow,ppwwyyxx/tensorflow,jalexvig/tensorflow,alshedivat/tensorflow,xzturn/tensorflow,cxxgtxy/tensorflow,adit-chandra/tensorflow,renyi533/tensorflow,gunan/tensorflow,rabipanda/tensorflow,Intel-tensorflow/tensorflow,aldian/tensorflow,caisq/tensorflow,adit-chandra/tensorflow,arborh/tensorflow,xzturn/tensorflow,aam-at/tensorflow,drpngx/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,karllessard/tensorflow,nburn42/tensorflow,freedomtan/tensorflow,brchiu/tensorflow,tensorflow/tensorflow-pywrap_saved_model,chemelnucfin/tensorflow,drpngx/tensorflow,Intel-tensorflow/tensorflow,petewarden/tensorflow,karllessard/tensorflow,gojira/tensorflow,nburn42/tensorflow,hfp/tensorflow-xsmm,xzturn/tensorflow,aselle/tensorflow,gojira/tensorflow,petewarden/tensorflow,jbedorf/tensorflow,girving/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,ageron/tensorflow,meteorcloudy/tensorflow,seanli9jan/tensorflow,rabipanda/tensorflow,seanli9jan/tensorflow,ZhangXinNan/tensorflow,av8ramit/tensorflow,kobejean/tensorflow,cxxgtxy/tensorflow,eaplatanios/tensorflow,nolanliou/tensorflow,yongtang/tensorflow,annarev/tensorflow,Intel-Corporation/tensorflow,jhseu/tensorflow,sarvex/tensorflow,aselle/tensorflow,arborh/tensorflow,jendap/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,gunan/tensorflow,Bismarrck/tensorflow,yanchen036/tensorflow,alshedivat/tensorflow,jbedorf/tensorflow,yongtang/tensorflow,av8ramit/tensorflow,aselle/tensorflow,Bismarrck/tensorflow,hfp/tensorflow-xsmm,alshedivat/tensorflow,nburn42/tensorflow,seanli9jan/tensorflow,hfp/tensorflow-xsmm,ghchinoy/tensorflow,manipopopo/tensorflow,yongtang/tensorflow,eaplatanios/tensorflow,petewarden/tensorflow,brchiu/tensorflow,ZhangXinNan/tensorflow,petewarden/tensorflow,jbedorf/tensorflow,theflofly/tensorflow,kevin-coder/tensorflow-fork,ZhangXinNan/tensorflow,gautam1858/tensorflow,frreiss/tensorflow-fred,ageron/tensorflow,xzturn/tensorflow,nburn42/tensorflow,zasdfgbnm/tensorflow,aselle/tensorflow,jart/tensorflow,dongjoon-hyun/tensorflow,meteorcloudy/tensorflow,tensorflow/tensorflow,nburn42/tensorflow,rabipanda/tensorflow,ghchinoy/tensorflow,aselle/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,davidzchen/tensorflow,eaplatanios/tensorflow,dongjoon-hyun/tensorflow,ZhangXinNan/tensorflow,apark263/tensorflow,AnishShah/tensorflow,tensorflow/tensorflow-pywrap_saved_model,allenlavoie/tensorflow,nburn42/tensorflow,nolanliou/tensorflow,brchiu/tensorflow,allenlavoie/tensorflow,tensorflow/tensorflow,Intel-Corporation/tensorflow,Intel-tensorflow/tensorflow,aselle/tensorflow,ravindrapanda/tensorflow,xzturn/tensorflow,meteorcloudy/tensorflow,xodus7/tensorflow,petewarden/tensorflow,DavidNorman/tensorflow,xzturn/tensorflow,meteorcloudy/tensorflow,rabipanda/tensorflow,asimshankar/tensorflow,ageron/tensorflow,kevin-coder/tensorflow-fork,ageron/tensorflow,zasdfgbnm/tensorflow,Xeralux/tensorflow,alshedivat/tensorflow,jalexvig/tensorflow,yongtang/tensorflow,yongtang/tensorflow,adit-chandra/tensorflow,nolanliou/tensorflow,Xeralux/tensorflow,freedomtan/tensorflow,Intel-Corporation/tensorflow,petewarden/tensorflow,girving/tensorflow,manipopopo/tensorflow,lukeiwanski/tensorflow,zasdfgbnm/tensorflow,alsrgv/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,jart/tensorflow,brchiu/tensorflow,aldian/tensorflow,allenlavoie/tensorflow,av8ramit/tensorflow,davidzchen/tensorflow,jart/tensorflow,freedomtan/tensorflow,manipopopo/tensorflow,AnishShah/tensorflow,yanchen036/tensorflow,jendap/tensorflow,jendap/tensorflow,apark263/tensorflow,aldian/tensorflow,paolodedios/tensorflow,frreiss/tensorflow-fred,ravindrapanda/tensorflow,gojira/tensorflow,davidzchen/tensorflow,dongjoon-hyun/tensorflow,xodus7/tensorflow,asimshankar/tensorflow,snnn/tensorflow,manipopopo/tensorflow,meteorcloudy/tensorflow,eaplatanios/tensorflow,allenlavoie/tensorflow,alshedivat/tensorflow,jhseu/tensorflow,hehongliang/tensorflow,jbedorf/tensorflow,arborh/tensorflow,av8ramit/tensorflow,DavidNorman/tensorflow,dancingdan/tensorflow,gunan/tensorflow,tensorflow/tensorflow,DavidNorman/tensorflow,av8ramit/tensorflow,nburn42/tensorflow,lukeiwanski/tensorflow,davidzchen/tensorflow,annarev/tensorflow,manipopopo/tensorflow,ppwwyyxx/tensorflow,allenlavoie/tensorflow,av8ramit/tensorflow,karllessard/tensorflow,allenlavoie/tensorflow,xzturn/tensorflow,aselle/tensorflow,benoitsteiner/tensorflow-xsmm,Bismarrck/tensorflow,freedomtan/tensorflow,Intel-Corporation/tensorflow,frreiss/tensorflow-fred,jhseu/tensorflow,lukeiwanski/tensorflow,ravindrapanda/tensorflow,dendisuhubdy/tensorflow,ghchinoy/tensorflow,ghchinoy/tensorflow,hehongliang/tensorflow,aam-at/tensorflow,zasdfgbnm/tensorflow,frreiss/tensorflow-fred,annarev/tensorflow,Xeralux/tensorflow,nolanliou/tensorflow,renyi533/tensorflow,rabipanda/tensorflow,Intel-tensorflow/tensorflow,freedomtan/tensorflow,arborh/tensorflow,Intel-Corporation/tensorflow,ppwwyyxx/tensorflow,jalexvig/tensorflow,davidzchen/tensorflow,hfp/tensorflow-xsmm,cxxgtxy/tensorflow,alsrgv/tensorflow,AnishShah/tensorflow,gojira/tensorflow,jendap/tensorflow,sarvex/tensorflow,girving/tensorflow,yongtang/tensorflow,AnishShah/tensorflow,yanchen036/tensorflow,frreiss/tensorflow-fred,jalexvig/tensorflow,asimshankar/tensorflow,jbedorf/tensorflow,ageron/tensorflow,kobejean/tensorflow,dendisuhubdy/tensorflow,aam-at/tensorflow,arborh/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,jhseu/tensorflow,Intel-tensorflow/tensorflow,chemelnucfin/tensorflow,gojira/tensorflow,ppwwyyxx/tensorflow,renyi533/tensorflow,gunan/tensorflow,cxxgtxy/tensorflow,DavidNorman/tensorflow,gojira/tensorflow,theflofly/tensorflow,manipopopo/tensorflow,dendisuhubdy/tensorflow,ageron/tensorflow,gautam1858/tensorflow,sarvex/tensorflow,annarev/tensorflow,jart/tensorflow,davidzchen/tensorflow,ghchinoy/tensorflow,asimshankar/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,Bismarrck/tensorflow,annarev/tensorflow,renyi533/tensorflow,AnishShah/tensorflow,girving/tensorflow,rabipanda/tensorflow,arborh/tensorflow,eaplatanios/tensorflow,yanchen036/tensorflow,renyi533/tensorflow,karllessard/tensorflow,frreiss/tensorflow-fred,Xeralux/tensorflow,seanli9jan/tensorflow,girving/tensorflow,kevin-coder/tensorflow-fork,zasdfgbnm/tensorflow,aam-at/tensorflow,allenlavoie/tensorflow,theflofly/tensorflow,aselle/tensorflow,jalexvig/tensorflow,xzturn/tensorflow,manipopopo/tensorflow,brchiu/tensorflow,zasdfgbnm/tensorflow,dancingdan/tensorflow,kevin-coder/tensorflow-fork,allenlavoie/tensorflow,ZhangXinNan/tensorflow,meteorcloudy/tensorflow,zasdfgbnm/tensorflow,tensorflow/tensorflow-pywrap_saved_model,hfp/tensorflow-xsmm,hehongliang/tensorflow,petewarden/tensorflow,jbedorf/tensorflow,jhseu/tensorflow,alshedivat/tensorflow,AnishShah/tensorflow,theflofly/tensorflow,meteorcloudy/tensorflow,benoitsteiner/tensorflow-xsmm,jalexvig/tensorflow,aldian/tensorflow,ghchinoy/tensorflow,benoitsteiner/tensorflow-xsmm,dancingdan/tensorflow,DavidNorman/tensorflow,gunan/tensorflow,lukeiwanski/tensorflow,xodus7/tensorflow,caisq/tensorflow,tensorflow/tensorflow-pywrap_saved_model,zasdfgbnm/tensorflow,lukeiwanski/tensorflow,dancingdan/tensorflow,DavidNorman/tensorflow,renyi533/tensorflow,chemelnucfin/tensorflow,rabipanda/tensorflow,aselle/tensorflow,aam-at/tensorflow,yanchen036/tensorflow,asimshankar/tensorflow,ageron/tensorflow,ghchinoy/tensorflow,snnn/tensorflow,petewarden/tensorflow,asimshankar/tensorflow,tensorflow/tensorflow-pywrap_saved_model,hehongliang/tensorflow,arborh/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,AnishShah/tensorflow,tensorflow/tensorflow,brchiu/tensorflow,eaplatanios/tensorflow,jbedorf/tensorflow,tensorflow/tensorflow,meteorcloudy/tensorflow,drpngx/tensorflow,Intel-Corporation/tensorflow,Bismarrck/tensorflow,paolodedios/tensorflow,apark263/tensorflow,drpngx/tensorflow,lukeiwanski/tensorflow,apark263/tensorflow,jhseu/tensorflow,chemelnucfin/tensorflow,jalexvig/tensorflow,renyi533/tensorflow,apark263/tensorflow,dongjoon-hyun/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,hfp/tensorflow-xsmm,seanli9jan/tensorflow,benoitsteiner/tensorflow-xsmm,caisq/tensorflow,ZhangXinNan/tensorflow,annarev/tensorflow,dendisuhubdy/tensorflow,paolodedios/tensorflow,ppwwyyxx/tensorflow,chemelnucfin/tensorflow,ravindrapanda/tensorflow,ppwwyyxx/tensorflow,eaplatanios/tensorflow,lukeiwanski/tensorflow,cxxgtxy/tensorflow,nolanliou/tensorflow,apark263/tensorflow,benoitsteiner/tensorflow-xsmm,alshedivat/tensorflow,hfp/tensorflow-xsmm,tensorflow/tensorflow-pywrap_saved_model,karllessard/tensorflow,kevin-coder/tensorflow-fork,caisq/tensorflow,snnn/tensorflow,aselle/tensorflow,kevin-coder/tensorflow-fork,karllessard/tensorflow,caisq/tensorflow,DavidNorman/tensorflow,caisq/tensorflow,hfp/tensorflow-xsmm,jalexvig/tensorflow,annarev/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,sarvex/tensorflow,seanli9jan/tensorflow,snnn/tensorflow,nburn42/tensorflow,dongjoon-hyun/tensorflow,dancingdan/tensorflow,jendap/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,alsrgv/tensorflow,kobejean/tensorflow,jalexvig/tensorflow,alsrgv/tensorflow,Bismarrck/tensorflow,alsrgv/tensorflow,Bismarrck/tensorflow,gautam1858/tensorflow,av8ramit/tensorflow,tensorflow/tensorflow,theflofly/tensorflow,arborh/tensorflow,drpngx/tensorflow,hehongliang/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,chemelnucfin/tensorflow,apark263/tensorflow,benoitsteiner/tensorflow-xsmm,aam-at/tensorflow,meteorcloudy/tensorflow,jbedorf/tensorflow,freedomtan/tensorflow,tensorflow/tensorflow,dongjoon-hyun/tensorflow,ppwwyyxx/tensorflow,snnn/tensorflow,dancingdan/tensorflow,allenlavoie/tensorflow,jendap/tensorflow,tensorflow/tensorflow-pywrap_saved_model,kevin-coder/tensorflow-fork,arborh/tensorflow,jhseu/tensorflow,rabipanda/tensorflow,dancingdan/tensorflow,dongjoon-hyun/tensorflow,tensorflow/tensorflow-pywrap_saved_model,dancingdan/tensorflow,dongjoon-hyun/tensorflow,Intel-tensorflow/tensorflow,frreiss/tensorflow-fred,girving/tensorflow,drpngx/tensorflow,AnishShah/tensorflow,aldian/tensorflow,karllessard/tensorflow,ageron/tensorflow,gunan/tensorflow,ageron/tensorflow,theflofly/tensorflow,allenlavoie/tensorflow,tensorflow/tensorflow-pywrap_saved_model,ppwwyyxx/tensorflow,theflofly/tensorflow,Bismarrck/tensorflow,jart/tensorflow,arborh/tensorflow,ZhangXinNan/tensorflow,sarvex/tensorflow,ravindrapanda/tensorflow,benoitsteiner/tensorflow-xsmm,jbedorf/tensorflow,Xeralux/tensorflow,theflofly/tensorflow,girving/tensorflow,paolodedios/tensorflow,dancingdan/tensorflow,ghchinoy/tensorflow,gautam1858/tensorflow,theflofly/tensorflow,aam-at/tensorflow,caisq/tensorflow,AnishShah/tensorflow,theflofly/tensorflow,alsrgv/tensorflow,gautam1858/tensorflow,drpngx/tensorflow,ageron/tensorflow,aam-at/tensorflow,gojira/tensorflow,jart/tensorflow,gunan/tensorflow,adit-chandra/tensorflow,snnn/tensorflow,annarev/tensorflow,rabipanda/tensorflow,snnn/tensorflow,hehongliang/tensorflow,paolodedios/tensorflow,davidzchen/tensorflow,aselle/tensorflow,frreiss/tensorflow-fred,frreiss/tensorflow-fred,jhseu/tensorflow,snnn/tensorflow,DavidNorman/tensorflow,hehongliang/tensorflow,xodus7/tensorflow,jart/tensorflow,Xeralux/tensorflow,DavidNorman/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,rabipanda/tensorflow,caisq/tensorflow,kobejean/tensorflow,nburn42/tensorflow,apark263/tensorflow,chemelnucfin/tensorflow,gautam1858/tensorflow,gojira/tensorflow,asimshankar/tensorflow,alsrgv/tensorflow,nolanliou/tensorflow,girving/tensorflow,petewarden/tensorflow,Intel-tensorflow/tensorflow,paolodedios/tensorflow,freedomtan/tensorflow,snnn/tensorflow,alsrgv/tensorflow,kevin-coder/tensorflow-fork,kobejean/tensorflow,theflofly/tensorflow,gojira/tensorflow,aldian/tensorflow,gunan/tensorflow,av8ramit/tensorflow,ppwwyyxx/tensorflow,DavidNorman/tensorflow,yongtang/tensorflow,alsrgv/tensorflow,nolanliou/tensorflow,Intel-tensorflow/tensorflow,zasdfgbnm/tensorflow,freedomtan/tensorflow,annarev/tensorflow,gautam1858/tensorflow,dongjoon-hyun/tensorflow,jalexvig/tensorflow,jendap/tensorflow