ó V¾÷Xc @`sËdZddlmZddlmZddlmZddlZddlZddlmZddl m Z ddl m Zd d d d d ddddg Zd„Zdefd„ƒYZd„Zdd„Zed„Zd„Zded„Zeeed„Zd„Zeeed„Zd„Zd„Z defd„ƒYZ!d „Z"d!„Z#d"„Z$ddd#„Z%dd$„Z&ddd%„Z'ej(d&ƒZ)d'„Z*d(d(d)„Z+d(d(d*„Z,dS(+s'Utility funtions for the graph_editor. i(tabsolute_import(tdivision(tprint_functionN(t iteritems(tops(t array_opstmake_list_of_opt get_tensorstmake_list_of_ttget_generating_opstget_consuming_opstControlOutputstplaceholder_nametmake_placeholder_from_tensort%make_placeholder_from_dtype_and_shapecC`s1x*|D]"}||kr|j|ƒqqW|S(s?Add all the elements of lb in la if they are not there already.(tappend(tlatlbtl((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pytconcatenate_unique)s  tListViewcB`sJeZdZd„Zd„Zd„Zd„ZeZd„Zd„Z RS(sYImmutable list wrapper. This class is strongly inspired by the one in tf.Operation. cC`s:t|tƒs-tdjt|ƒƒƒ‚n||_dS(NsExpected a list, got: {}.(t isinstancetlistt TypeErrortformatttypet_list(tselftlist_((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pyt__init__8scC`s t|jƒS(N(titerR(R((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pyt__iter__=scC`s t|jƒS(N(tlenR(R((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pyt__len__@scC`s t|jƒS(N(tboolR(R((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pyt__bool__CscC`s |j|S(N(R(Rti((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pyt __getitem__IscC`s,t|tƒst|ƒ}nt|ƒ|S(N(RR(Rtother((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pyt__add__Ls( t__name__t __module__t__doc__RRR!R#t __nonzero__R%R'(((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pyR2s     cC`s)yt|ƒ}Wntk r$tSXtS(s&Return true if the object is iterable.(Rt ExceptiontFalsetTrue(tobjt_((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pyt is_iterableSs  cC`s|dkrg}nt|tƒrQxdt|ƒD]\}}t||ƒq1Wn:t|ƒr~x+|D]}t||ƒqdWn |j|ƒ|S(sFlatten a tree into a list. Args: tree: iterable or not. If iterable, its elements (child) can also be iterable or not. leaves: list to which the tree leaves are appended (None by default). Returns: A list of all the leaves in the tree. N(tNoneRtdictRt flatten_treeR1R(ttreetleavesR0tchild((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pyR4\s     c`s>t|ƒr0t|tƒrW|jt|ƒƒ}|j‡fd†t|ƒDƒƒ|St|tƒrËt|dƒrŸ|jt|ƒt |j ƒˆƒ}n(|jt|ƒ‡fd†|Dƒƒ}|St|t j ƒr|jt|ƒƒ}|j‡fd†|Dƒƒ|S|‡fd†|DƒƒSn ˆ|ƒSdS(sµTransform all the nodes of a tree. Args: tree: iterable or not. If iterable, its elements (child) can also be iterable or not. fn: function to apply to each leaves. iterable_type: type use to construct the resulting tree for unknwon iterable, typically `list` or `tuple`. Returns: A tree whose leaves has been transformed by `fn`. The hierarchy of the output tree mimics the one of the input tree. c3`s*|] \}}|t|ˆƒfVqdS(N(ttransform_tree(t.0tkR7(tfn(sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pys „st_asdictc3`s|]}t|ˆƒVqdS(N(R8(R9R7(R;(sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pys Œsc3`s|]}t|ˆƒVqdS(N(R8(R9R7(R;(sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pys sc3`s|]}t|ˆƒVqdS(N(R8(R9R7(R;(sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pys “sN( R1RR3t__new__RRRttuplethasattrR8R<t collectionstSequence(R5R;t iterable_typetres((R;sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pyR8ss$ *cG`s„d}xwt|ƒD]i\}}|dkrF|jdk rF|j}q|jdk r|j|k rtdj|ƒƒ‚qqWdS(sÎCheck that all the element in args belong to the same graph. Args: *args: a list of object with a obj.graph property. Raises: ValueError: if all the elements do not belong to the same graph. sArgument[{}]: Wrong graph!N(R2t enumeratetgrapht ValueErrorR(targsRER$tsgv((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pyt check_graphs˜s  cC`sGt|tjƒr|St|ƒs@tdjt|ƒƒƒ‚n|dkratjtj f}nt|ƒsy|f}nd}xŸ|D]—}t||ƒsÞtdjdj g|D]}t |ƒ^q±ƒt|ƒƒƒ‚n|dkrö|j }q†||j k r†t dj|ƒƒ‚q†q†W|dkrC| rCt dƒ‚n|S(s8Return the unique graph used by the all the elements in tops. Args: tops: list of elements to check (usually a list of tf.Operation and/or tf.Tensor). Or a tf.Graph. check_types: check that the element in tops are of given type(s). If None, the types (tf.Operation, tf.Tensor) are used. none_if_empty: don't raise an error if tops is an empty list, just return None. Returns: The unique graph used by all the tops. Raises: TypeError: if tops is not a iterable of tf.Operation. ValueError: if the graph is not unique. s{} is not iterables Expected a type in ({}), got: {}s, s+Operation {} does not belong to given graphs,Can't find the unique graph of an empty listN(Rttf_opstGraphR1RRRR2t OperationtTensortjointstrRERF(ttopst check_typest none_if_emptytgtoptt((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pytget_unique_graph¨s(     1  cC`s®t|tjƒr1|r"|jƒStdƒ‚nyt|ƒsI|g}n|sSgS|r|redntj}t|d|ƒng|D]}t|tjƒrˆ|^qˆSdS(s2Convert ops to a list of `tf.Operation`. Args: ops: can be an iterable of `tf.Operation`, a `tf.Graph` or a single operation. check_graph: if `True` check if all the operations belong to the same graph. allow_graph: if `False` a `tf.Graph` cannot be converted. ignore_ts: if True, silently ignore `tf.Tensor`. Returns: A newly created list of `tf.Operation`. Raises: TypeError: if ops cannot be converted to a list of `tf.Operation` or, if `check_graph` is `True`, if all the ops do not belong to the same graph. s0allow_graph is False: cannot convert a tf.Graph.RQN( RRJRKtget_operationsRR1R2RLRV(Rt check_grapht allow_grapht ignore_tsRQRT((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pyRÎs   cC`s^t|tjƒs0tdjt|ƒƒƒ‚ng}x!|jƒD]}||j7}qCW|S(sÆget all the tensors which are input or output of an op in the graph. Args: graph: a `tf.Graph`. Returns: A list of `tf.Tensor`. Raises: TypeError: if graph is not a `tf.Graph`. sExpected a graph, got: {}(RRJRKRRRRWtoutputs(REttsRT((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pyRïs cC`s®t|tjƒr1|r"t|ƒStdƒ‚nyt|ƒsI|g}n|sSgS|r|redntj}t|d|ƒng|D]}t|tjƒrˆ|^qˆSdS(sConvert ts to a list of `tf.Tensor`. Args: ts: can be an iterable of `tf.Tensor`, a `tf.Graph` or a single tensor. check_graph: if `True` check if all the tensors belong to the same graph. allow_graph: if `False` a `tf.Graph` cannot be converted. ignore_ops: if `True`, silently ignore `tf.Operation`. Returns: A newly created list of `tf.Tensor`. Raises: TypeError: if `ts` cannot be converted to a list of `tf.Tensor` or, if `check_graph` is `True`, if all the ops do not belong to the same graph. s0allow_graph is False: cannot convert a tf.Graph.RQN( RRJRKRRR1R2RMRV(R\RXRYt ignore_opsRQRU((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pyRs   cC`s,t|dtƒ}g|D]}|j^qS(sReturn all the generating ops of the tensors in `ts`. Args: ts: a list of `tf.Tensor` Returns: A list of all the generating `tf.Operation` of the tensors in `ts`. Raises: TypeError: if `ts` cannot be converted to a list of `tf.Tensor`. RY(RR-RT(R\RU((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pyR s cC`s`t|dtƒ}g}xA|D]9}x0|jƒD]"}||kr2|j|ƒq2q2WqW|S(sûReturn all the consuming ops of the tensors in ts. Args: ts: a list of `tf.Tensor` Returns: A list of all the consuming `tf.Operation` of the tensors in `ts`. Raises: TypeError: if ts cannot be converted to a list of `tf.Tensor`. RY(RR-t consumersR(R\RRURT((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pyR -s   cB`sJeZdZd„Zd„Zd„Zd„Zd„Zed„ƒZ RS(sThe control outputs topology.cC`sYt|tjƒs0tdjt|ƒƒƒ‚ni|_||_d|_ |j ƒdS(seCreate a dictionary of control-output dependencies. Args: graph: a `tf.Graph`. Returns: A dictionary where a key is a `tf.Operation` instance and the corresponding value is a list of all the ops which have the key as one of their control-input dependencies. Raises: TypeError: graph is not a `tf.Graph`. sExpected a tf.Graph, got: {}N( RRJRKRRRt_control_outputst_graphR2t_versiont_build(RRE((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pyRCs    cC`s&|j|jjkr"|jƒn|S(s4Update the control outputs if the graph has changed.(RaR`tversionRb(R((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pytupdateVs cC`s|jjƒ|jjƒ}xk|D]c}xZ|jD]O}||jkrXg|j|R8RIR-RVR.RRRR R R RmRqRrR R RtcompileR‚R†RR”(((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/util.pytsP    !  % &!   7   %  $