[XLA] Make XlaAllocator obey retry_on_failure arg.
[XLA] Make XlaAllocator obey retry_on_failure arg.

Previously we ignored it.

PiperOrigin-RevId: 195908178

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