Update tests to use the new distribute_py_test build rule.
Update tests to use the new distribute_py_test build rule.

PiperOrigin-RevId: 229634446

gunan/tensorflow,gautam1858/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,chemelnucfin/tensorflow,yongtang/tensorflow,tensorflow/tensorflow-pywrap_saved_model,Intel-tensorflow/tensorflow,ppwwyyxx/tensorflow,paolodedios/tensorflow,theflofly/tensorflow,xzturn/tensorflow,jbedorf/tensorflow,Intel-tensorflow/tensorflow,tensorflow/tensorflow-pywrap_saved_model,ppwwyyxx/tensorflow,aldian/tensorflow,apark263/tensorflow,freedomtan/tensorflow,apark263/tensorflow,renyi533/tensorflow,adit-chandra/tensorflow,petewarden/tensorflow,aam-at/tensorflow,davidzchen/tensorflow,Intel-Corporation/tensorflow,jendap/tensorflow,davidzchen/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,annarev/tensorflow,Intel-tensorflow/tensorflow,adit-chandra/tensorflow,alsrgv/tensorflow,yongtang/tensorflow,annarev/tensorflow,arborh/tensorflow,tensorflow/tensorflow-pywrap_saved_model,Intel-tensorflow/tensorflow,jbedorf/tensorflow,ageron/tensorflow,DavidNorman/tensorflow,alsrgv/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,renyi533/tensorflow,gunan/tensorflow,aam-at/tensorflow,freedomtan/tensorflow,aldian/tensorflow,paolodedios/tensorflow,aam-at/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,Intel-tensorflow/tensorflow,tensorflow/tensorflow,petewarden/tensorflow,tensorflow/tensorflow,xzturn/tensorflow,yongtang/tensorflow,chemelnucfin/tensorflow,ppwwyyxx/tensorflow,sarvex/tensorflow,alsrgv/tensorflow,cxxgtxy/tensorflow,alsrgv/tensorflow,xzturn/tensorflow,annarev/tensorflow,DavidNorman/tensorflow,ghchinoy/tensorflow,freedomtan/tensorflow,jendap/tensorflow,adit-chandra/tensorflow,arborh/tensorflow,aldian/tensorflow,aldian/tensorflow,gunan/tensorflow,tensorflow/tensorflow-pywrap_saved_model,petewarden/tensorflow,cxxgtxy/tensorflow,paolodedios/tensorflow,kevin-coder/tensorflow-fork,tensorflow/tensorflow,tensorflow/tensorflow,chemelnucfin/tensorflow,ghchinoy/tensorflow,jhseu/tensorflow,freedomtan/tensorflow,chemelnucfin/tensorflow,yongtang/tensorflow,adit-chandra/tensorflow,aam-at/tensorflow,tensorflow/tensorflow-pywrap_saved_model,aam-at/tensorflow,tensorflow/tensorflow,tensorflow/tensorflow,DavidNorman/tensorflow,jhseu/tensorflow,yongtang/tensorflow,apark263/tensorflow,frreiss/tensorflow-fred,annarev/tensorflow,cxxgtxy/tensorflow,Intel-tensorflow/tensorflow,jbedorf/tensorflow,sarvex/tensorflow,jendap/tensorflow,davidzchen/tensorflow,karllessard/tensorflow,gunan/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,paolodedios/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,aam-at/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,ppwwyyxx/tensorflow,aam-at/tensorflow,yongtang/tensorflow,xzturn/tensorflow,adit-chandra/tensorflow,karllessard/tensorflow,chemelnucfin/tensorflow,gunan/tensorflow,kevin-coder/tensorflow-fork,ageron/tensorflow,Intel-tensorflow/tensorflow,ageron/tensorflow,theflofly/tensorflow,arborh/tensorflow,ageron/tensorflow,kevin-coder/tensorflow-fork,arborh/tensorflow,ageron/tensorflow,davidzchen/tensorflow,theflofly/tensorflow,renyi533/tensorflow,adit-chandra/tensorflow,DavidNorman/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,jhseu/tensorflow,adit-chandra/tensorflow,davidzchen/tensorflow,jhseu/tensorflow,Intel-Corporation/tensorflow,adit-chandra/tensorflow,cxxgtxy/tensorflow,annarev/tensorflow,gautam1858/tensorflow,xzturn/tensorflow,renyi533/tensorflow,karllessard/tensorflow,aldian/tensorflow,theflofly/tensorflow,karllessard/tensorflow,xzturn/tensorflow,arborh/tensorflow,chemelnucfin/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,Intel-Corporation/tensorflow,chemelnucfin/tensorflow,yongtang/tensorflow,adit-chandra/tensorflow,theflofly/tensorflow,gunan/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,DavidNorman/tensorflow,frreiss/tensorflow-fred,tensorflow/tensorflow-pywrap_tf_optimizer,paolodedios/tensorflow,sarvex/tensorflow,apark263/tensorflow,Intel-tensorflow/tensorflow,alsrgv/tensorflow,jbedorf/tensorflow,freedomtan/tensorflow,DavidNorman/tensorflow,sarvex/tensorflow,cxxgtxy/tensorflow,ghchinoy/tensorflow,freedomtan/tensorflow,apark263/tensorflow,ppwwyyxx/tensorflow,Intel-Corporation/tensorflow,tensorflow/tensorflow,arborh/tensorflow,ghchinoy/tensorflow,alsrgv/tensorflow,davidzchen/tensorflow,chemelnucfin/tensorflow,DavidNorman/tensorflow,jhseu/tensorflow,petewarden/tensorflow,gautam1858/tensorflow,theflofly/tensorflow,jhseu/tensorflow,annarev/tensorflow,xzturn/tensorflow,gunan/tensorflow,petewarden/tensorflow,alsrgv/tensorflow,frreiss/tensorflow-fred,tensorflow/tensorflow-experimental_link_static_libraries_once,paolodedios/tensorflow,apark263/tensorflow,ghchinoy/tensorflow,karllessard/tensorflow,chemelnucfin/tensorflow,jbedorf/tensorflow,jhseu/tensorflow,freedomtan/tensorflow,ghchinoy/tensorflow,Intel-tensorflow/tensorflow,jbedorf/tensorflow,sarvex/tensorflow,ppwwyyxx/tensorflow,DavidNorman/tensorflow,Intel-tensorflow/tensorflow,aam-at/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,davidzchen/tensorflow,gautam1858/tensorflow,kevin-coder/tensorflow-fork,ageron/tensorflow,renyi533/tensorflow,annarev/tensorflow,ppwwyyxx/tensorflow,petewarden/tensorflow,aldian/tensorflow,apark263/tensorflow,apark263/tensorflow,ghchinoy/tensorflow,jbedorf/tensorflow,theflofly/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,kevin-coder/tensorflow-fork,arborh/tensorflow,karllessard/tensorflow,renyi533/tensorflow,renyi533/tensorflow,frreiss/tensorflow-fred,apark263/tensorflow,adit-chandra/tensorflow,tensorflow/tensorflow-pywrap_saved_model,gautam1858/tensorflow,xzturn/tensorflow,gautam1858/tensorflow,paolodedios/tensorflow,renyi533/tensorflow,gunan/tensorflow,tensorflow/tensorflow,renyi533/tensorflow,gautam1858/tensorflow,paolodedios/tensorflow,gautam1858/tensorflow,alsrgv/tensorflow,kevin-coder/tensorflow-fork,karllessard/tensorflow,tensorflow/tensorflow,frreiss/tensorflow-fred,theflofly/tensorflow,chemelnucfin/tensorflow,jhseu/tensorflow,gunan/tensorflow,aam-at/tensorflow,ageron/tensorflow,yongtang/tensorflow,Intel-Corporation/tensorflow,ppwwyyxx/tensorflow,jendap/tensorflow,freedomtan/tensorflow,arborh/tensorflow,freedomtan/tensorflow,freedomtan/tensorflow,xzturn/tensorflow,davidzchen/tensorflow,tensorflow/tensorflow-pywrap_saved_model,davidzchen/tensorflow,Intel-Corporation/tensorflow,ageron/tensorflow,aam-at/tensorflow,gunan/tensorflow,renyi533/tensorflow,jbedorf/tensorflow,davidzchen/tensorflow,Intel-Corporation/tensorflow,sarvex/tensorflow,gautam1858/tensorflow,aam-at/tensorflow,alsrgv/tensorflow,freedomtan/tensorflow,frreiss/tensorflow-fred,ageron/tensorflow,aam-at/tensorflow,alsrgv/tensorflow,DavidNorman/tensorflow,petewarden/tensorflow,annarev/tensorflow,tensorflow/tensorflow-pywrap_saved_model,kevin-coder/tensorflow-fork,tensorflow/tensorflow-pywrap_tf_optimizer,jendap/tensorflow,Intel-Corporation/tensorflow,renyi533/tensorflow,jendap/tensorflow,cxxgtxy/tensorflow,petewarden/tensorflow,yongtang/tensorflow,annarev/tensorflow,annarev/tensorflow,adit-chandra/tensorflow,petewarden/tensorflow,davidzchen/tensorflow,jendap/tensorflow,tensorflow/tensorflow-pywrap_saved_model,petewarden/tensorflow,jendap/tensorflow,apark263/tensorflow,sarvex/tensorflow,paolodedios/tensorflow,ghchinoy/tensorflow,ppwwyyxx/tensorflow,karllessard/tensorflow,tensorflow/tensorflow-pywrap_saved_model,kevin-coder/tensorflow-fork,paolodedios/tensorflow,yongtang/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,frreiss/tensorflow-fred,theflofly/tensorflow,arborh/tensorflow,karllessard/tensorflow,gautam1858/tensorflow,frreiss/tensorflow-fred,alsrgv/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,jendap/tensorflow,davidzchen/tensorflow,petewarden/tensorflow,jhseu/tensorflow,chemelnucfin/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,freedomtan/tensorflow,ageron/tensorflow,aldian/tensorflow,ageron/tensorflow,DavidNorman/tensorflow,ghchinoy/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,karllessard/tensorflow,theflofly/tensorflow,gunan/tensorflow,jhseu/tensorflow,jbedorf/tensorflow,karllessard/tensorflow,theflofly/tensorflow,DavidNorman/tensorflow,frreiss/tensorflow-fred,apark263/tensorflow,frreiss/tensorflow-fred,xzturn/tensorflow,gautam1858/tensorflow,jendap/tensorflow,aldian/tensorflow,chemelnucfin/tensorflow,renyi533/tensorflow,theflofly/tensorflow,yongtang/tensorflow,xzturn/tensorflow,ppwwyyxx/tensorflow,gautam1858/tensorflow,annarev/tensorflow,paolodedios/tensorflow,ppwwyyxx/tensorflow,ghchinoy/tensorflow,ageron/tensorflow,ppwwyyxx/tensorflow,xzturn/tensorflow,jendap/tensorflow,tensorflow/tensorflow-pywrap_saved_model,ghchinoy/tensorflow,kevin-coder/tensorflow-fork,alsrgv/tensorflow,jbedorf/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,DavidNorman/tensorflow,arborh/tensorflow,frreiss/tensorflow-fred,petewarden/tensorflow,sarvex/tensorflow,jbedorf/tensorflow,jhseu/tensorflow,adit-chandra/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,jbedorf/tensorflow,cxxgtxy/tensorflow,kevin-coder/tensorflow-fork,arborh/tensorflow,ghchinoy/tensorflow,tensorflow/tensorflow,tensorflow/tensorflow,kevin-coder/tensorflow-fork,arborh/tensorflow,gunan/tensorflow,frreiss/tensorflow-fred,tensorflow/tensorflow-experimental_link_static_libraries_once,Intel-tensorflow/tensorflow,jhseu/tensorflow,cxxgtxy/tensorflow