Always cast `tf.distributions.Distribution` `_event_shape`, `_batch_shape`.
Always cast `tf.distributions.Distribution` `_event_shape`, `_batch_shape`.

PiperOrigin-RevId: 190415923

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