o ®Ÿ?e“ ã@srdZddlmZGdd„deƒZddd„Zd d „Zd d „Zd d„Zea dd„Z dd„Z dd„Z dd„Z dd„ZdS)zGradient tape utilities.é)Ú pywrap_tfec@s&eZdZdZdgZdd„Zdd„ZdS)ÚTapez(Represents a gradient propagation trace.Ú_tapecCs ||_dS©N)r)ÚselfÚtape©rú]/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/eager/tape.pyÚ__init__s z Tape.__init__cCs t |j¡Sr)rZTFE_Py_TapeWatchedVariablesr)rrrr Úwatched_variabless zTape.watched_variablesN)Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú __slots__r r rrrr rs  rFTcCst ||¡}t|ƒS)z&Pushes a new tape onto the tape stack.)rZTFE_Py_TapeSetNewr)Ú persistentZwatch_accessed_variablesrrrr Ú push_new_tape s rcCót |j¡dS)z,Pushes an existing tape onto the tape stack.N)rZTFE_Py_TapeSetAddr©rrrr Ú push_tape&órcCst |j|¡dS)z2Marks this tensor to be watched by the given tape.N)rZTFE_Py_TapeWatchr)rZtensorrrr Úwatch+srcCs|gSrr)Úvariablerrr Údefault_get_variables0srcCsttusJ‚|adS)zºRegisters the resolver to be used to get the list of variables to watch. Args: resolver: callable, takes a Variable and returns a list of Variables that shall be watched. N)Ú_variables_overrider)Úresolverrrr Ú"register_watched_variable_resolver9s rcCs.t|ƒ}|D]}t |j|¡t |¡qdS)z4Marks this variable to be watched by the given tape.N)rrZTFE_Py_TapeWatchVariablerÚ&TFE_Py_VariableWatcherVariableAccessed)rrÚ variablesÚvarrrr Úwatch_variableEs  þr cCs*t|ƒ}|D] }t |¡t |¡qdS)ztNotifies all tapes in the stack that a variable has been accessed. Args: variable: variable to be watched. N)rrÚTFE_Py_TapeVariableAccessedr)rrrrrr Úvariable_accessedMs   þr"cCsDg}|D] }|jr| t|ƒ¡q|D] }t |¡t |¡qdS)z¼Notifies all tapes in the stack that variables have been accessed. Only trainable variables are marked as accessed. Args: variables: iterable of variables to mark as accessed. N)Z trainableÚextendrrr!r)rZaccessedrrrrr Úvariables_accessedYs€  þr$cCr)z!Pops the given tape in the stack.N)rZTFE_Py_TapeSetRemoverrrrr Úpop_tapekrr%N)FT)rZtensorflow.pythonrÚobjectrrrrrrrr r"r$r%rrrr Ús