[tf.data] Fixing an error in the optimization loop.
[tf.data] Fixing an error in the optimization loop.

PiperOrigin-RevId: 213386401

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