BVh fdZddlmZGddeZddZdZdZdZea d Z d Z d Z d Z d Zy)zGradient tape utilities.) pywrap_tfec"eZdZdZdgZdZdZy)Tapez(Represents a gradient propagation trace._tapec||_yN)r)selftapes L/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/eager/tape.py__init__z Tape.__init__s DJc@tj|jSr)rTFE_Py_TapeWatchedVariablesr)r s r watched_variableszTape.watched_variabless  1 1$** ==r N)__name__ __module__ __qualname____doc__ __slots__r rr r rrs0i)>r rcDtj||}t|S)z&Pushes a new tape onto the tape stack.)rTFE_Py_TapeSetNewr) persistentwatch_accessed_variablesr s r push_new_taper s  % %j2J K$ dr cBtj|jy)z,Pushes an existing tape onto the tape stack.N)rTFE_Py_TapeSetAddrr s r push_taper&s tzz*r cDtj|j|y)z2Marks this tensor to be watched by the given tape.N)rTFE_Py_TapeWatchr)r tensors r watchr#+s djj&1r c|gSrr)variables r default_get_variablesr&0s r c$ttusJ|ay)zRegisters 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&)resolvers r "register_watched_variable_resolverr*9s  5 55 5 r ct|}|D]7}tj|j|tj|9y)z4Marks this variable to be watched by the given tape.N)r(rTFE_Py_TapeWatchVariabler&TFE_Py_VariableWatcherVariableAccessed)r r% variablesvars r watch_variabler0Es=!(+) ;c'' C855c:;r c|t|}|D],}tj|tj|.y)ztNotifies all tapes in the stack that a variable has been accessed. Args: variable: variable to be watched. N)r(rTFE_Py_TapeVariableAccessedr-)r%r.r/s r variable_accessedr3Ms9 "(+) ;c**3/55c:;r cg}|D])}|js|jt|+|D],}tj|tj |.y)zNotifies 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) trainableextendr(rr2r-)r.accessedr%r/s r variables_accessedr8Ys`(5hoo)(345;c**3/55c:;r cBtj|jy)z!Pops the given tape in the stack.N)rTFE_Py_TapeSetRemoverrs r pop_taper;ks !!$**-r N)FT)rtensorflow.pythonrobjectrrrr#r&r(r*r0r3r8r;rr r r>sN( >6 > + 2  , !; ;;$.r