B d@s dZddlZGdddeZdS)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) cs"tt|d|_t|_dS)N)superr__init___session threadingLock_initialization_lock)self) __class__O/var/www/html/venv/lib/python3.7/site-packages/tensorboard/util/op_evaluator.pyr>szPersistentOpEvaluator.__init__c Csvddlmm}|jT|jr$dS|}||WdQRX|jddid}|j ||d|_WdQRXdS)z@Initialize the graph and session, if this has not yet been done.rNZGPU)Z device_count)graphconfig) Ztensorflow.compat.v1compatZv1rrZGraph as_defaultinitialize_graphZ ConfigProtoSession)r tfr rr r r _lazily_initializeCs z(PersistentOpEvaluator._lazily_initializecCs tddS)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".N)NotImplementedError)r r r r rRsz&PersistentOpEvaluator.initialize_graphcOs tddS)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".N)r)r argskwargsr r r run[s zPersistentOpEvaluator.runc Os*||j|j||SQRXdS)N)rrrr)r rrr r r __call__fs zPersistentOpEvaluator.__call__) __name__ __module__ __qualname____doc__rrrrr __classcell__r r )r r rs $   r)rrobjectrr r r r s