o #jh5@sddlZddlZddlmZddlmZmZddlmZddl m Z ddl m Z ddl mZmZmZmZmZmZgZd d Zd d Zed ZedZedZddZ dddZddZddZddZdddZ ddZ!dS) N)StringIO)_C_ops _legacy_C_ops)check_variable_and_dtype) framework_pb2)Variable) LayerHelper OpProtoHolderconvert_np_dtype_to_dtype_corein_dynamic_modein_dynamic_or_pir_modecCs tdd|}tdd|S)a  Formatting. Args: name: The name/alias This function takes in a name and converts it to a standard format of group1_group2. Where as per the regular expression, group1 can have alphabets and numbers and group2 has capital alphabets. z(.)([A-Z][a-z]+)z\1_\2z([a-z0-9])([A-Z]))resublower)names1rg/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/tensor/layer_function_generator.py _convert_$s rcCs tj|SN)rZAttrTypeName)tprrr _type_to_str_4s rz\$\$([^\$]+)\$\$z \$([^\$]+)\$z !!([^!]+)!!cCs td|S)Nz :math:`\1`)_two_dollar_pattern_r)textrrr escape_math=s rc Cst|tjs tdt}|t|j|d|jD]1}dt |j }|||d|t|j|j rA|d|j rI|d|dqt }|d|d |d |rn|D]}||qf|jD]2}|j |vryqq|d||j |d |t|j|d |t|j|dqq|d ur|D]} | } |d|| |dqt|jdkr|d|d|jD]} | jsnq|t | j |d|t| j|S)z Generate docstring by OpProto Args: op_proto (framework_pb2.OpProto): a protobuf message typed OpProto Returns: str: the document string z)OpProto should be `framework_pb2.OpProto`z Args: z z (Tensor): z Duplicatable.z Optional. Z use_mkldnnZis_testZ use_cudnnz (z): Nrz Returns: ) isinstancerZOpProto TypeErrorrwritercommentinputsrr duplicableZ dispensabler Zgenerated_op_attr_namesaddattrsrtypestriplenoutputs intermediategetvalue) op_protoadditional_args_linesZskip_attrs_setbuf each_inputZ line_beginZ skip_attrst each_attrlineeach_optrrr_generate_doc_string_Esb                          r5cstddjD}ddjD}t|dkr%tdd|djr.td|D] }|jr:td d q0|djd d|Dfd d fdd}|_t |_ |S)zRegister the Python layer for an Operator. Args: op_type: The name of the operator to be created. This function takes in the operator type (sigmoid, mean , average etc) and creates the operator functionality. cSsg|]}|js|qSrr+.0outputrrr z%generate_layer_fn..cSsg|]}|jr|qSrr6r7rrrr:r;z0Only one non intermediate output operator can bezautomatically generated. rz6Only non duplicable op can be automatically generated.z0The op can be automatically generated only when z(all intermediate ops are not duplicable.cSsg|]}|jqSr)rr7rrrr:sc sd}|jD][}t|j}||g}t|tst|ts|g}t|dkr7t|dkr,q|dg}|dd}|D]&}t|tsHt dd|durP|j }q9||j kr_t d ||j q9q|dur| d}|r}t|t jjsyt|}|S|}|St jjj}|S)z^ This function performs the sanity check for dtype and instance type. Nrr<z input of z must be variablez+operator {} must input same dtype. {} vs {}dtype)r#rrpoprlisttupler)r ValueErrorr=formatgetr ZVarDescZVarTyper ZFP32) r-argskwargsr=iptrvalZeachZ arg_dtypeop_typerrinfer_and_check_dtypesD           z0generate_layer_fn..infer_and_check_dtypec stfi|}g|Ri|}i}jD]5}t|j}||g}t|ts2t|ts2|g}t|dkrHt|dkrH|d}|dd}|||j<qi}|tg} | rht| ttfre| dn| } n|j |d} | g|<D] }|j |dg||<qu|j |||d| | S)Nrr<r=)r'r#r*r&) r r#rrr>rr?r@r)"create_variable_for_type_inference append_opZappend_activation) rDrEhelperr=r#rFrrGr*outZout_varrJZintermediate_output_namesZo_namer-rIrrfuncs2          zgenerate_layer_fn..func) r instance get_op_protor*r)rAr$r__name__r5__doc__)rIZnot_intermediate_outputsZintermediate_outputsr9rQrrPrgenerate_layer_fns<     ) rVcsHt}dfdd }|_dkrd|_|St|dgd|_|S) zRegister the Python layer for an Operator without Attribute. Args: op_type: The name of the operator to be created. This function takes in the operator type (sigmoid, exp , tanh etc) and creates the operator functionality. Ncstrttrtt}||Stt}||Sdvr(t|dgdn t|dgdtfit}|j|j d}|j d|id|id|S) N)absexpZsquarex)float16float32float64)Zint32Zint64rZr[r\Z complex64Z complex128Zuint16rKXZOut)r'r#r*) rhasattrrgetattrrrr localsrLr=rM)rYroprNr9rHrrrQ s,     z$generate_activation_fn..funcrWa> Abs Operator. Perform elementwise abs for input `X`. .. math:: out = |x| Args: x (Tensor): The input tensor of abs op. out (Tensor): The output tensor of abs op. name (str, optional): Name for the operation (optional, default is None). For more information, please refer to :ref:`api_guide_Name`. zname (str, optional): Name for the operation (optional, default is None). For more information, please refer to :ref:`api_guide_Name`.)r.r)r rRrSrTrUr5)rIr-rQrrHrgenerate_activation_fns )rbcs2dd}dfdd }|_d|||_|S)aRegister the Python layer for an Inplace Operator without Attribute. Args: inplace_op_type: The name of the inplace operator to be created. This function takes in the inplace operator type (exp_ , ceil_ etc) and creates the operator functionality. Ncs8trttrtt}||Stt}||SdSr)r r^rr_r)rYrrainplace_op_typerrrQ[s   z!generate_inplace_fn..funcz{ Inplace version of ``{}`` API, the output Tensor will be inplaced with input ``x``. Please refer to :ref:`api_paddle_{}`. r)rTrBrU)reZorigin_op_typerQrrdrgenerate_inplace_fnPs  rfcsddfdd}|S)a Decorator of layer function. It will use the docstring from the layer function as the template. The template arguments are: * ${comment}: The operator comment written in CPP. * ${{name}_comment}: The comment of ${name} written with AddAttr, AddOutput, and AddInput. The ${name} is Python snake style. i.e., xxx_xxx. * ${{name}_type}: The type of ${name}. Returns: Decorated function. cSs |dS)N.)rstrip)msgrrrtrim_ending_dot|s z$templatedoc..trim_ending_dotc sPdur|j}n}t|}t|j}|jd}d}|D]!}| }t |dkr8|t |7}|d7}q!t |dkrB|d7}q!d|i}|j D]}t |j} |j|| d<d|| d <qL|jD]} t | j} | j|| d<t| j|| d <qh|jD]} t | j} | j|| d<d|| d <q|||_|S) Nrr z r"_commentr_type)rTr rRrSstringTemplaterUr"splitr(r)rr#rrr&rr'r* substitute) rQZ op_type_namer-tmplZ comment_linesr"r3rDr0Z input_namer2r4Z output_namerIrjrr__impl__s<              ztemplatedoc..__impl__r)rIrurrtr templatedocns#rvcCs|j||_dS)z Append sample code for dynamically generated functions. Args: func: The function of the function to be append sample code to. sample_code: sample code session in rst format. N)rU)rQZ sample_coderrradd_sample_codesrw)NNr)"rroiorZpaddlerrZbase.data_feederrZ base.protorZcommon_ops_importrZ frameworkr r r r r r__all__rrcompilerZ_single_dollar_pattern_Z_two_bang_pattern_rr5rVrbrfrvrwrrrrs,          GuO  7