BVhb dZddlZddlZddlmZGddej dddgZdZejd Z y) z}Utilities for managing forward accumulators. A separate file from forwardprop.py so that functions can use these utilities. N) pywrap_tfec$eZdZdZdfd ZxZS) TangentInfozFPacked forward accumulator state. The return value of `pack_tangents`.c>|d}|g}tt| |||S)N)superr__new__)clsindicestangents __class__s X/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/eager/forwardprop_util.pyr zTangentInfo.__new__s.gh c *3 BB)NN)__name__ __module__ __qualname____doc__r __classcell__)r s@rrrsNCCrrr r c8ttj|S)a$Packs forward accumulator state into a TangentInfo tuple. Args: tensors: A flat list of Tensors to pack forward accumulator state for. Returns: A tuple of (indices, tangents): indices: A sequence of sequences of two-element tuples. Each forward accumulator is represented as a sequence of tuples with (primal_index, jvp_index). Both integers index into the concatenated `tensors + jvps` array. tangents: A flat list of Tensors. Best interpreted as a sequence to be appended to `tensors`. )rrTFE_Py_PackJVPs)tensorss r pack_tangentsr&s j009 ::rc#K tjdtjy#tjwxYww)aTemporarily push or pop transient state for accumulators in the active set. Allows an accumulator which is currently processing an operation to temporarily reset its state. This is useful when building forwardprop versions of functions, where an accumulator will trigger function building and then must process captured symbolic tensors while building it. Without pushing and popping, accumulators ignore operations executed as a direct result of their own jvp computations. Yields: None (used for its side effect). N)r"TFE_Py_ForwardAccumulatorPushState!TFE_Py_ForwardAccumulatorPopStaterrrpush_forwardprop_stater8s23113 002J002sA 1A AA ) r collections contextlibtensorflow.pythonr namedtuplerrcontextmanagerrrrrr"sW ( CK=9j*AB C;$ 33r