B ѻd4#@sdZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z GdddeZd d Zd d Zd dZddZiZddZdddZdddZdS)zFDecorator that provides a warning if the wrapped object is never used.N)context)ops) tf_logging) tf_decoratorc@s0eZdZdZddZddZddZdd Zd S) _TFShouldUseHelpera5Object stored in TFShouldUse-wrapped objects. When it is deleted it will emit a warning or error if its `sate` method has not been called by time of deletion, and Tensorflow is not executing eagerly or inside a tf.function (which use autodeps and resolve the main issues this wrapper warns about). csb|_|_|_|_tr*| _n4trX|rPd_t fddq^d_nd_dS)NFcs jddS)NT) raise_error) _check_sated)selfr V/var/www/html/venv/lib/python3.7/site-packages/tensorflow/python/util/tf_should_use.py2z-_TFShouldUseHelper.__init__..T) _type_repr _stack_frameZ_error_in_functionrexecuting_eagerly_satedrinside_function'add_exit_callback_to_default_func_graph)r type_repr_ stack_frameerror_in_function warn_in_eagerr )r r __init__%s z_TFShouldUseHelper.__init__cCs"d|_d|_d|_d|_d|_dS)NT)rrrrZ_logging_module)r r r r sate9s z_TFShouldUseHelper.satecCsr|jr dSdddtj|jddD}|rVztd|j|j|Wd| Xnt d|j|j|dS) z#Check if the object has been sated.NcSsg|] }|qSr )rstrip).0liner r r Esz3_TFShouldUseHelper._check_sated..)limitzObject was never used (type {}): {}. If you want to mark it as used call its "mark_used()" method. It was originally created here: {}z================================== Object was never used (type {}): {} If you want to mark it as used call its "mark_used()" method. It was originally created here: {} ==================================) rjoin traceback format_stackr RuntimeErrorformatrrrrerror)r rZcreation_stackr r r r@s z_TFShouldUseHelper._check_satedcCs|jdddS)NF)r)r)r r r r __del__Xsz_TFShouldUseHelper.__del__N)__name__ __module__ __qualname____doc__rrrr)r r r r rs rcCs||_||_dS)N)_tf_should_use_helper_tf_should_use_wrapped_value)r wrapped_valuetf_should_use_helperr r r _new__init__\sr2cCs*|dkrt|||Stt|d||S)N)r.r/r/)object __setattr__setattr__getattribute__)r keyvaluer r r _new__setattr__bs  r9cCs>|dkrt|d|dkr,t||Stt|d|S)N)r.r/r.)r. mark_usedZ __setatt__r/)r3r6rgetattr)r r7r r r _new__getattribute__js  r<cOsHt|dytt|dd}|||Stk rBYnXdS)Nr.r/r:)r3r6rAttributeError)r argskwargsmur r r _new_mark_usedss  rAcst|}t|d}|r"|||St|tdj}tjdkrdfdd}t j j ||d}ntj |t j }t|_t|_t|_t|_|t|<|||S)aVCreate a wrapper for object x, whose class subclasses type(x). The wrapper will emit a warning if it is deleted without any of its properties being accessed or methods being called. Args: x: The instance to wrap. tf_should_use_helper: The object that tracks usage. Returns: An object wrapping `x`, of type `type(x)`. N__orig_bases__)r!cs|j|S)N)update__dict__)ns)txr r set_bodys z_get_wrapper..set_body) exec_body)type _WRAPPERSgetcopydeepcopyr; __bases__sys version_infotypes new_classr*dictrEr2rr<r6rAr:r9r4)xr1Ztype_xZmemoizedbasesrHZcopy_txr )rGr _get_wrappers      rWFcCs|dkst|tr|s|Str*|s*|Str:|s:|Sy tWn$tk rhtdj j }YnXt t |t ||||d}t||S)a Wraps object x so that if it is never used, a warning is logged. Args: x: Python object. error_in_function: Python bool. If `True`, a `RuntimeError` is raised if the returned value is never used when created during `tf.function` tracing. warn_in_eager: Python bool. If `True` raise warning if in Eager mode as well as graph mode. Returns: An instance of `TFShouldUseWarningWrapper` which subclasses `type(x)` and is a very shallow wrapper for `x` which logs access into `x`. N)rrrrr) isinstancelistrrrr ValueErrorrPexc_infotb_framef_backrrJreprrW)rUrrrr1r r r _add_should_use_warnings"   r`cs&fdd}|dk r||S|SdS)aFunction wrapper that ensures the function's output is used. If the output is not used, a `logging.error` is logged. If `error_in_function` is set, then a `RuntimeError` will be raised at the end of function tracing if the output is not used by that point. An output is marked as used if any of its attributes are read, modified, or updated. Examples when the output is a `Tensor` include: - Using it in any capacity (e.g. `y = t + 0`, `sess.run(t)`) - Accessing a property (e.g. getting `t.name` or `t.op`). - Calling `t.mark_used()`. Note, certain behaviors cannot be tracked - for these the object may not be marked as used. Examples include: - `t != 0`. In this case, comparison is done on types / ids. - `isinstance(t, tf.Tensor)`. Similar to above. Args: fn: The function to wrap. warn_in_eager: Whether to create warnings in Eager as well. error_in_function: Whether to raise an error when creating a tf.function. Returns: The wrapped function. csnfdd}jpd}|dd}t|dkr8|}n|\}}d|t|g}d}tj|d||dS) zDecorates the input function.cst||dS)N)rr)r`)r>r?)rfnrr r wrappeds z5should_use_result..decorated..wrappedr z Note: The output of this function should be used. If it is not, a warning will be logged or an error may be raised. To mark the output as used, call its .mark_used() method.should_use_result)targetdecorator_funcdecorator_name decorator_doc)r-splitlenr#textwrapdedentrmake_decorator)rarbZfn_docZ split_docZ updated_docZbriefrestnote)rr)rar decorateds   z$should_use_result..decoratedNr )rarrrqr )rrr resre)FF)NFF)r-rMrPrlr$rRtensorflow.python.eagerrtensorflow.python.frameworkrtensorflow.python.platformrtensorflow.python.utilrr3rr2r9r<rArKrWr`rer r r r s$    @  * )