o "j7@sddlZddlZddlZddlmZddlmZmZddlm Z ddl m Z m Z m Z mZmZddlmZddlmZgZd d Zd d Zed ZedZedZddZ d!ddZddZddZddZd"ddZ d#ddZ!dd Z"dS)$N)StringIO)_C_ops _legacy_C_ops)check_variable_and_dtype) OpProtoHolderVariableconvert_np_dtype_to_dtype_corein_dygraph_mode) LayerHelper) framework_pb2cCs 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)names1rl/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/base/layers/layer_function_generator.py _convert_$s rcCs tj|SN)r ZAttrTypeName)tprrr _type_to_str_1s 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: ) isinstancer ZOpProto TypeErrorrwritercommentinputsrr duplicableZ dispensablerZgenerated_op_attr_namesaddattrsrtypestriplenoutputs intermediategetvalue) op_protoadditional_args_linesZskip_attrs_setbuf each_inputZ line_beginZ skip_attrst each_attrlineeach_optrrr_generate_doc_string_Bsb                          r4cstddjD}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|qSrr5r6rrrr9r: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)rr6rrrr9sc 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) rCrDhelperr<r"rErrFr)outZout_varrIZintermediate_output_namesZo_namer,rHrrfuncs2          zgenerate_layer_fn..func) rinstance get_op_protor)r(r@r#r__name__r4__doc__)rHZnot_intermediate_outputsZintermediate_outputsr8rPrrOrgenerate_layer_fns<    ) rUcs6t}dfdd }|_t|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}||Str"ttr"tt}||Sdvr0t|dgdn t|dgdtfit}|j|j d}|j d|id|id|S) N)absexpZsquarex)float16float32float64uint16)Zint32Zint64rYrZr[Z complex64Z complex128r\rJXZOut)r&r"r)) r hasattrrgetattrrrr localsrKr<rL)rXroprMr8rGrrrPs(    z$generate_activation_fn..funczname (str, optional): Name for the operation (optional, default is None). For more information, please refer to :ref:`api_guide_Name`.)r-r)rrQrRrSr4rT)rHr,rPrrGrgenerate_activation_fns%rbcs4dddfdd }|_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. Ncsttr tt}||Stdddlm}|r3t|dr3|j r3t d|j |j |j t ||S)NzPIn static mode, {}() is the same as {}() and does not perform inplace operation.r)in_to_static_mode is_view_varaSorry about what's happend. In to_static mode, {}'s output variable {} is a viewed Tensor in dygraph. This will result in inconsistent calculation behavior between dynamic and static graphs. You must find the location of the strided API be called, and call {} = {}.assign().)r r_rwarningswarnrAZ dygraph.baserdr^rer@rZnamebrb)rXrrardinplace_op_typeZorigin_op_typerrrP: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_base_layers_{}`. r)rSrArT)rirPrrhrgenerate_inplace_fn1s rjcsfdd}|S)Ncstt|j|_|Sr)r4rrQrRrSrT)rPr!rr__impl__^szautodoc..__impl__r)r!rmrrlrautodoc]s  rncsddfdd}|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_dotvs 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) Nrrkr z r!_commentr_type)rSrrQrRstringTemplaterTr!splitr'r(rr"rrr%rr&r) substitute) rPZ op_type_namer,tmplZ comment_linesr!r2rCr/Z input_namer1r3Z output_namerHrrrrrmys<              ztemplatedoc..__impl__r)rHrmrr{r templatedocjs #r|cCs|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)rT)rPZ sample_coderrradd_sample_codesr})NN)rkr)#rrvrfiorZpaddlerrZ data_feederrZ frameworkrrr r r Z layer_helperr protor __all__rrcompilerZ_single_dollar_pattern_Z_two_bang_pattern_rr4rUrbrjrnr|r}rrrrs0        Er8 , 5