Convert Softmax to custom op
Convert Softmax to custom op

Change the old TF2XLA bridge to convert a SoftMax op to a custom call to
softmax when the custom_fake_quant_op_calls flag is enabled. This is to
preserve the information for quantization.

PiperOrigin-RevId: 305892297
Change-Id: Id372e7da5753dd3109d6e993c6193d155ba136bd

Intel-Corporation/tensorflow,tensorflow/tensorflow-pywrap_saved_model,cxxgtxy/tensorflow,gunan/tensorflow,annarev/tensorflow,tensorflow/tensorflow,gautam1858/tensorflow,Intel-Corporation/tensorflow,petewarden/tensorflow,gautam1858/tensorflow,gunan/tensorflow,yongtang/tensorflow,Intel-tensorflow/tensorflow,freedomtan/tensorflow,karllessard/tensorflow,tensorflow/tensorflow,karllessard/tensorflow,petewarden/tensorflow,davidzchen/tensorflow,tensorflow/tensorflow-pywrap_saved_model,karllessard/tensorflow,Intel-Corporation/tensorflow,aam-at/tensorflow,freedomtan/tensorflow,aam-at/tensorflow,aam-at/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,freedomtan/tensorflow,tensorflow/tensorflow-pywrap_saved_model,karllessard/tensorflow,davidzchen/tensorflow,tensorflow/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,freedomtan/tensorflow,gautam1858/tensorflow,gunan/tensorflow,yongtang/tensorflow,gautam1858/tensorflow,cxxgtxy/tensorflow,aam-at/tensorflow,aldian/tensorflow,gunan/tensorflow,karllessard/tensorflow,yongtang/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,yongtang/tensorflow,frreiss/tensorflow-fred,frreiss/tensorflow-fred,aldian/tensorflow,aam-at/tensorflow,paolodedios/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,tensorflow/tensorflow-pywrap_tf_optimizer,petewarden/tensorflow,Intel-Corporation/tensorflow,gautam1858/tensorflow,paolodedios/tensorflow,karllessard/tensorflow,Intel-tensorflow/tensorflow,karllessard/tensorflow,annarev/tensorflow,annarev/tensorflow,paolodedios/tensorflow,tensorflow/tensorflow-pywrap_saved_model,davidzchen/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,frreiss/tensorflow-fred,annarev/tensorflow,yongtang/tensorflow,frreiss/tensorflow-fred,annarev/tensorflow,gunan/tensorflow,gautam1858/tensorflow,tensorflow/tensorflow,davidzchen/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,paolodedios/tensorflow,karllessard/tensorflow,cxxgtxy/tensorflow,tensorflow/tensorflow-pywrap_saved_model,tensorflow/tensorflow-experimental_link_static_libraries_once,tensorflow/tensorflow-pywrap_saved_model,gunan/tensorflow,cxxgtxy/tensorflow,frreiss/tensorflow-fred,annarev/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,petewarden/tensorflow,tensorflow/tensorflow,gautam1858/tensorflow,aam-at/tensorflow,petewarden/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,gunan/tensorflow,Intel-tensorflow/tensorflow,petewarden/tensorflow,yongtang/tensorflow,yongtang/tensorflow,davidzchen/tensorflow,gunan/tensorflow,gautam1858/tensorflow,frreiss/tensorflow-fred,davidzchen/tensorflow,tensorflow/tensorflow,tensorflow/tensorflow,aldian/tensorflow,davidzchen/tensorflow,freedomtan/tensorflow,Intel-Corporation/tensorflow,Intel-Corporation/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,aam-at/tensorflow,sarvex/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,gautam1858/tensorflow,frreiss/tensorflow-fred,petewarden/tensorflow,aldian/tensorflow,petewarden/tensorflow,freedomtan/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,freedomtan/tensorflow,yongtang/tensorflow,paolodedios/tensorflow,davidzchen/tensorflow,Intel-tensorflow/tensorflow,tensorflow/tensorflow-pywrap_saved_model,freedomtan/tensorflow,Intel-tensorflow/tensorflow,karllessard/tensorflow,frreiss/tensorflow-fred,paolodedios/tensorflow,cxxgtxy/tensorflow,Intel-tensorflow/tensorflow,tensorflow/tensorflow-pywrap_saved_model,tensorflow/tensorflow,Intel-Corporation/tensorflow,tensorflow/tensorflow,sarvex/tensorflow,yongtang/tensorflow,annarev/tensorflow,petewarden/tensorflow,petewarden/tensorflow,frreiss/tensorflow-fred,petewarden/tensorflow,gautam1858/tensorflow,cxxgtxy/tensorflow,aam-at/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,aam-at/tensorflow,yongtang/tensorflow,tensorflow/tensorflow-pywrap_saved_model,tensorflow/tensorflow-pywrap_tf_optimizer,annarev/tensorflow,paolodedios/tensorflow,aam-at/tensorflow,aam-at/tensorflow,Intel-Corporation/tensorflow,davidzchen/tensorflow,tensorflow/tensorflow,tensorflow/tensorflow,yongtang/tensorflow,frreiss/tensorflow-fred,tensorflow/tensorflow-pywrap_saved_model,cxxgtxy/tensorflow,Intel-tensorflow/tensorflow,aldian/tensorflow,gunan/tensorflow,davidzchen/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,gautam1858/tensorflow,aldian/tensorflow,sarvex/tensorflow,frreiss/tensorflow-fred,cxxgtxy/tensorflow,tensorflow/tensorflow-pywrap_saved_model,davidzchen/tensorflow,frreiss/tensorflow-fred,paolodedios/tensorflow,aldian/tensorflow,gunan/tensorflow,karllessard/tensorflow,aldian/tensorflow,petewarden/tensorflow,davidzchen/tensorflow,paolodedios/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,freedomtan/tensorflow,tensorflow/tensorflow-experimental_link_static_libraries_once,freedomtan/tensorflow,annarev/tensorflow,annarev/tensorflow,gunan/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,tensorflow/tensorflow-experimental_link_static_libraries_once,annarev/tensorflow,gunan/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,Intel-tensorflow/tensorflow,Intel-tensorflow/tensorflow,sarvex/tensorflow,gautam1858/tensorflow,tensorflow/tensorflow-pywrap_tf_optimizer,sarvex/tensorflow,Intel-tensorflow/tensorflow,freedomtan/tensorflow,paolodedios/tensorflow,sarvex/tensorflow,paolodedios/tensorflow,karllessard/tensorflow,Intel-tensorflow/tensorflow,sarvex/tensorflow,sarvex/tensorflow,aam-at/tensorflow,freedomtan/tensorflow