o 2h@sdZddlZGdddZdS)zFTensorBoard helper routine for TF op evaluator. Requires TensorFlow. Ncs@eZdZdZfddZddZddZdd Zd d ZZ S) PersistentOpEvaluatoraEvaluate a fixed TensorFlow graph repeatedly, safely, efficiently. Extend this class to create a particular kind of op evaluator, like an image encoder. In `initialize_graph`, create an appropriate TensorFlow graph with placeholder inputs. In `run`, evaluate this graph and return its result. This class will manage a singleton graph and session to preserve memory usage, and will ensure that this graph and session do not interfere with other concurrent sessions. A subclass of this class offers a threadsafe, highly parallel Python entry point for evaluating a particular TensorFlow graph. Example usage: class FluxCapacitanceEvaluator(PersistentOpEvaluator): """Compute the flux capacitance required for a system. Arguments: x: Available power input, as a `float`, in jigawatts. Returns: A `float`, in nanofarads. """ def initialize_graph(self): self._placeholder = tf.placeholder(some_dtype) self._op = some_op(self._placeholder) def run(self, x): return self._op.eval(feed_dict: {self._placeholder: x}) evaluate_flux_capacitance = FluxCapacitanceEvaluator() for x in xs: evaluate_flux_capacitance(x) cstd|_t|_dSN)super__init___session threadingLock_initialization_lockself __class__X/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorboard/util/op_evaluator.pyr>s zPersistentOpEvaluator.__init__c Csddlmm}|jA|jr WddS|}| |Wdn1s0wY|jddid}|j ||d|_WddS1sPwYdS)z@Initialize the graph and session, if this has not yet been done.rNGPU) device_count)graphconfig) tensorflow.compat.v1compatv1r rGraph as_defaultinitialize_graph ConfigProtoSession)r tfrrrrr_lazily_initializeCs  "z(PersistentOpEvaluator._lazily_initializecCtd)zCreate the TensorFlow graph needed to compute this operation. This should write ops to the default graph and return `None`. z-Subclasses must implement "initialize_graph".NotImplementedErrorr rrrrRsz&PersistentOpEvaluator.initialize_graphcOr)aEvaluate the ops with the given input. When this function is called, the default session will have the graph defined by a previous call to `initialize_graph`. This function should evaluate any ops necessary to compute the result of the query for the given *args and **kwargs, likely returning the result of a call to `some_op.eval(...)`. z Subclasses must implement "run".rr argskwargsrrrrun[s zPersistentOpEvaluator.runcOsF||j|j|i|WdS1swYdSr)rrrr$r!rrr__call__fs $zPersistentOpEvaluator.__call__) __name__ __module__ __qualname____doc__rrrr$r% __classcell__rrr rrs %  r)r)rrrrrrs