BVhi dZddlZddlZddlZddlmZedd5Zeje5ddl m Z dddddddZ dZ e dk(rejZej!d d d ej#d ed ddej#deddej#deddej'\Z Zej*e ej,dgezyy#1swYxYw#1swYxYw)abPrints a header file to be used with SELECTIVE_REGISTRATION. An example of command-line usage is: bazel build tensorflow/python/tools:print_selective_registration_header && \ bazel-bin/tensorflow/python/tools/print_selective_registration_header \ --graphs=path/to/graph.pb > ops_to_register.h Then when compiling tensorflow, include ops_to_register.h in the include search path and pass -DSELECTIVE_REGISTRATION and -DSUPPORT_SELECTIVE_REGISTRATION - see core/framework/selective_registration.h for more details. When compiling for Android: bazel build -c opt --copt="-DSELECTIVE_REGISTRATION" \ --copt="-DSUPPORT_SELECTIVE_REGISTRATION" \ //tensorflow/tools/android/inference_interface:libtensorflow_inference.so \ --host_crosstool_top=@bazel_tools//tools/cpp:toolchain \ --crosstool_top=//external:android/crosstool --cpu=armeabi-v7a N)appz /dev/nullw)!selective_registration_header_libctjjd}tt j |tj tjy)N,)FLAGSgraphssplitprintr get_headerproto_fileformat default_ops) unused_argvr s k/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/tools/print_selective_registration_header.pymainr2sB <<  c "&'226383I3I383D3DFG__main__typeboolc(|jdk(S)Ntrue)lower)vs rr<sAGGI,?rz--graphsz>"]]".)rrrz --default_opsz#NoOp:NoOp,_Recv:RecvOp,_Send:SendOpa$Default operator:kernel pairs to always include implementation for.Pass "all" to have all operators and kernels included; note that this should be used only when it is useful compared with simply not using selective registration, as it can in some cases limit the effect of compilation caches)rargv)__doc__argparse contextlibsysabslropenfredirect_stdouttensorflow.python.toolsrrr__name__ArgumentParserparserregister add_argumentstrparse_known_argsunparsedrunr rrr4sE&  +sHq"<*"<"