[TF:XLA] Small performance tweaks for tf.random_shuffle, but still too slow.
[TF:XLA] Small performance tweaks for tf.random_shuffle, but still too slow.

PiperOrigin-RevId: 200086551

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