BVh dZddlZddlZddlZddlmZddlmZddlm Z ddlm Z ddl m Z e dgZd ZeeZd Zd Zd Zd ZdZdZ ddZy)zComputes a header file to be used with SELECTIVE_REGISTRATION. See the executable wrapper, print_selective_registration_header.py, for more information. N) text_format) graph_pb2)gfile) tf_logging)_pywrap_kernel_registry AccumulateNV2sFlexct}tj|dj}|s t dt j |}|D]\}}||r|ndf}|j| |S)z7Gets the ops and kernels needed from the ops list file.rzInput file should not be emptyN)setrGFileread Exceptionjsonloadsadd) input_fileops ops_list_strops_listopkernel op_and_kernels i/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/tools/selective_registration_header_lib.py_get_ops_from_ops_listr,ss #Z-224,  4 55 ZZ %(jb&6t4MGGM *ct}|jt|j|jj D]&}|jt|j (|S)z:Gets the ops and kernels needed from the tensorflow model.)r update_get_ops_from_nodedefsnodelibraryfunctionnode_def) graph_defrr!s r_get_ops_from_graphdefr$9sW #** #INN 34##,,:hJJ%h&7&789: *rc|jsd|_tj|j}t |j }|s|t vr!||rt |jdfSdfStjd|y)aGets the op and kernel needed from the given NodeDef. Args: node_def: TF NodeDef to get op/kernel information. Returns: A tuple of (op_name, kernel_name). If the op is not in the allowlist of ops without kernel and there is no kernel found, then return None. z/cpu:0zutf-8Nz"Warning: no kernel found for op %s) devicerTryFindKernelClassSerializeToStringstrrOPS_WITHOUT_KERNEL_ALLOWLISTdecoderwarning)r" kernel_classrs rget_ops_from_nodedefr.Cs HO(;;  "$, 8;;"R77 \L''01 LLt LL;R@ rcft}|D]!}t|}|s|j|#|S)afGets the ops and kernels needed from the list of NodeDef. If a NodeDef's op is not in the allowlist of ops without kernel and there is no kernel found for this NodeDef, then skip that NodeDef and proceed to the next one. Args: node_defs: list of NodeDef's to get op/kernel information. Returns: A set of (op_name, kernel_name) tuples. )r r.r) node_defsrr"rs rrrYs: #h(2M ggm *rc`t}|D]}tjd||dk(r|jt |}9t j |dj}|dk(r tjj|}n/|dk(sJtj|tj}|jt|}|rI|dk7rD|jdD]0}|jd\}} || f} | |vs |j| 2t!|S) z5Gets the ops and kernels needed from the model files.zLoading proto file %srrbrawproto textprotoall,:)r rinfounionrrr r rGraphDef FromStringrParser$splitrsorted) proto_fileformat proto_filesdefault_ops_strr proto_file file_datar#srrrs rget_ops_and_kernelsrEns #7jOO+Z8:% II,Z8 9c J-224I:%$$// :i  ,, ,##Iy/A/A/CDi ))*95 6C7"E1  " "3 '773z2get_header_from_ops_and_kernels..s ,52qB ,sc(j|yrH)append)rD result_lists rrNz/get_header_from_ops_and_kernels..appendsqrrz$// This file was autogenerated by %sz#ifndef OPS_TO_REGISTERz#define OPS_TO_REGISTERz##define SHOULD_REGISTER_OP(op) truez+#define SHOULD_REGISTER_OP_KERNEL(clz) truez(#define SHOULD_REGISTER_OP_GRADIENT truea namespace { constexpr const char* skip(const char* x) { return (*x) ? (*x == ' ' ? skip(x + 1) : x) : x; } constexpr bool isequal(const char* x, const char* y) { return (*skip(x) && *skip(y)) ? (*skip(x) == *skip(y) && isequal(skip(x) + 1, skip(y) + 1)) : (!*skip(x) && !*skip(y)); } template struct find_in { static constexpr bool f(const char* x, const char* const y[N]) { return isequal(x, y[0]) || find_in::f(x, y + 1); } }; template<> struct find_in<0> { static constexpr bool f(const char* x, const char* const y[]) { return false; } }; } // end namespace z6constexpr const char* kNecessaryOpKernelClasses[] = { z"%s", z};z#define SHOULD_REGISTER_OP_KERNEL(clz) (find_in::f(clz, kNecessaryOpKernelClasses))z9constexpr inline bool ShouldRegisterOp(const char op[]) {z return falsez || isequal(op, "%s")z ;}z3#define SHOULD_REGISTER_OP(op) ShouldRegisterOp(op)z$#define SHOULD_REGISTER_OP_GRADIENT SymbolicGradienttruefalsez#endif )r>r ospathr=sysargvjoin) ops_and_kernelsinclude_all_ops_and_kernelsrrNrK script_nameliner-rrOs @rget_header_from_ops_and_kernelsr_st?+/ ,O ,,#+77==!-.![ /+ => "# "#  01 89 56 D6 EED*'<   i,&&d' DLD 4L  )* 2J FG Sk/ (2 -./ 5M 3K @A 2J 1(C/FW >?  ; rcVt|||}|s tdyt||dk(S)aComputes a header for use with tensorflow SELECTIVE_REGISTRATION. Args: graphs: a list of paths to GraphDef files to include. proto_fileformat: optional format of proto file, either 'textproto', 'rawproto' (default) or ops_list. The ops_list is the file contain the list of ops in JSON format, Ex: "[["Transpose", "TransposeCpuOp"]]". default_ops: optional comma-separated string of operator:kernel pairs to always include implementation for. Pass 'all' to have all operators and kernels included. Default: 'NoOp:NoOp,_Recv:RecvOp,_Send:SendOp'. Returns: the string of the header that should be written as ops_to_register.h. zError reading graph!r5)rEprintr_)graphsr? default_opsr[s r get_headerres4"((8&+N/  ! (+:N OOr)r3z#NoOp:NoOp,_Recv:RecvOp,_Send:SendOp)__doc__rrVrXgoogle.protobufrtensorflow.core.frameworkrtensorflow.python.platformrrtensorflow.python.utilr frozensetr* FLEX_PREFIXlenFLEX_PREFIX_LENGTHrr$r.rrEr_rerIrrros{  '/,1: )*   %   , *>R l!+@Pr