ó V¾÷Xc @`sUdZddlmZddlmZddlmZddlmZddlmZddlm Z dd d d d d ddddg Z d„Z de fd„ƒYZd%d%d„Zd%d%d„Zd%d%d„Zd%d%d„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd „Zd!„Zd"„Zd#„Zd$„Zd%S(&s%Various function for graph rerouting.i(tabsolute_import(tdivision(tprint_function(tsubgraph(tutil(topstswap_tst reroute_tst swap_inputstreroute_inputst swap_outputstreroute_outputstswap_iost reroute_iostremove_control_inputstadd_control_inputscC`stj|ƒ}tj|ƒ}t|ƒt|ƒkr]tdjt|ƒt|ƒƒƒ‚nxt||ƒD]Œ\}}|j|j}}|j|ƒs¶tdj||ƒƒ‚n|jƒ|jƒ}}|j|ƒsmtdj||ƒƒ‚qmqmWdS(sIMake sure the shape and dtype of the two tensor's lists are compatible. Args: ts0: an object convertible to a list of `tf.Tensor`. ts1: an object convertible to a list of `tf.Tensor`. Raises: ValueError: if any pair of tensors (same index in ts0 and ts1) have a dtype or a shape which is not compatible. s*ts0 and ts1 have different sizes: {} != {}s$Dtypes {} and {} are not compatible.s$Shapes {} and {} are not compatible.N( Rtmake_list_of_ttlent ValueErrortformattziptdtypetis_compatible_witht get_shape(tts0tts1tt0tt1tdtype0tdtype1tshape0tshape1((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/reroute.pyt_check_ts_compatibility's    t _RerouteModecB`s2eZdZedƒ\ZZZed„ƒZRS(s1Enums for reroute's mode. swap: the end of tensors a and b are swapped. a2b: the end of the tensor a are also rerouted to the end of the tensor b (the end of b is left dangling). b2a: the end of the tensor b are also rerouted to the end of the tensor a (the end of a is left dangling). icC`sd||jkrttfS||jkr2ttfS||jkrKttfStdj|ƒƒ‚dS(sîCheck swap mode. Args: mode: an integer representing one of the modes. Returns: A tuple `(a2b, b2a)` boolean indicating what rerouting needs doing. Raises: ValueError: if mode is outside the enum range. sUnknown _RerouteMode: {}N(tswaptTruetb2atFalseta2bRR(tclstmode((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/reroute.pytcheckNs    ( t__name__t __module__t__doc__trangeR"R&R$t classmethodR)(((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/reroute.pyR!Csc C`sËd}|dk r||M}n|dk r8||8}ni}xF|D]>}gt|jƒD]\}} | |kr[|^q[||b" or "b<->a": the tensors' end are swapped. After this operation, the previous consumers of t0 are now consumers of t1 and vice-versa. 2) The reroute mode is "a->b": the tensors' end of t0 are re-routed to the tensors's end of t1 (which are left dangling). After this operation, the previous consumers of t0 are still consuming t0 but the previous consumers of t1 are not also consuming t0. The tensor t1 has no consumer. 3) The reroute mode is "b->a": this mode is the symmetric of the "a->b" mode. Note that this function is re-routing the end of two tensors, not the start. Re-routing the start of two tensors is not supported by this library. The reason for that is the following: TensorFlow, by design, creates a strong bond between an op and its output tensor. This Graph editor follows this design and treats an operation A and its generating tensors {t_i} as an entity which cannot be broken. In other words, an op cannot be detached from any of its output tensors, ever. But it is possible to detach an op from its input tensors, which is what this function concerns itself with. Warning: this function is directly manipulating the internals of the tf.Graph. Args: ts0: an object convertible to a list of `tf.Tensor`. ts1: an object convertible to a list of `tf.Tensor`. mode: what to do with those tensors: "a->b" or "b<->a" for swaping and "a->b" or "b->a" for one direction re-routing. can_modify: iterable of operations which can be modified. Any operation outside within_ops will be left untouched by this function. cannot_modify: iterable of operations which cannot be modified. Any operation within cannot_modify will be left untouched by this function. Returns: The number of individual modifications made by the function. Raises: TypeError: if `ts0` or `ts1` cannot be converted to a list of `tf.Tensor`. TypeError: if `can_modify` or `cannot_modify` is not `None` and cannot be converted to a list of `tf.Operation`. iN( R!R)RRR R/t frozensettmake_list_of_opRtsett consumerstappendR;(RRR(R4R5R&R$R6tprecomputed_consumersRRt consumers0R3R?((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/reroute.pyt _reroute_ts…s2-   "   cC`st||tj||ƒS(sFor each tensor's pair, swap the end of (t0,t1). B0 B1 B0 B1 | | => X A0 A1 A0 A1 Args: ts0: an object convertible to a list of `tf.Tensor`. ts1: an object convertible to a list of `tf.Tensor`. can_modify: iterable of operations which can be modified. Any operation outside within_ops will be left untouched by this function. cannot_modify: iterable of operations which cannot be modified. Any operation within cannot_modify will be left untouched by this function. Returns: The number of individual modifications made by the function. Raises: TypeError: if ts0 or ts1 cannot be converted to a list of tf.Tensor. TypeError: if can_modify or cannot_modify is not None and cannot be converted to a list of tf.Operation. (RCR!R"(RRR4R5((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/reroute.pyRÎscC`st||tj||ƒS(s\For each tensor's pair, replace the end of t1 by the end of t0. B0 B1 B0 B1 | | => |/ A0 A1 A0 A1 The end of the tensors in ts1 are left dangling. Args: ts0: an object convertible to a list of `tf.Tensor`. ts1: an object convertible to a list of `tf.Tensor`. can_modify: iterable of operations which can be modified. Any operation outside within_ops will be left untouched by this function. cannot_modify: iterable of operations which cannot be modified. Any operation within cannot_modify will be left untouched by this function. Returns: The number of individual modifications made by the function. Raises: TypeError: if ts0 or ts1 cannot be converted to a list of tf.Tensor. TypeError: if can_modify or cannot_modify is not None and cannot be converted to a list of tf.Operation. (RCR!R&(RRR4R5((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/reroute.pyRçscC`swtj|ƒ\}}t|tjƒsEtdjt|ƒƒƒ‚nt|tjƒsutdjt|ƒƒƒ‚ntj ||ƒ|j ƒ}|j ƒ}|rÞ|rÞ|j |j |_ |_ |j |j |_ |_ nF|r|j |_ |j |_ n#|r$|j |_ |j |_ nd„}|rC|||ƒn|rY|||ƒn|j |ƒ|j |ƒdS(s¯Remap in place the inputs of two subgraph views to mimic the reroute. This function is meant to used by reroute_inputs only. Args: sgv0: the first subgraph to have its inputs remapped. sgv1: the second subgraph to have its inputs remapped. mode: reroute mode, see _reroute_ts(...). Raises: TypeError: if svg0 or svg1 are not SubGraphView. ValueError: if sgv0 and sgv1 do not belong to the same graph. sExpected a SubGraphView, got {}cS`s\xUt|jƒD]D\}}||jkr|jj|ƒ}|j||j|Re-route both the inputs and the outputs of the two subgraph views. This involves swapping all the inputs/ouputs of the two subgraph views. Args: sgv0: the first subgraph to be swapped. This argument is converted to a subgraph using the same rules than the function subgraph.make_view. sgv1: the second subgraph to be swapped. This argument is converted to a subgraph using the same rules than the function subgraph.make_view. mode: reroute mode, see _reroute_ts(...). Returns: A tuple `(sgv0, sgv1)` of subgraph views with their outputs and inputs swapped. Note that the function argument sgv0 and sgv1 are also modified in place. Raises: StandardError: if sgv0 or sgv1 cannot be converted to a SubGraphView using the same rules than the function subgraph.make_view. (R]R[(RSRTR(((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/reroute.pyt _reroute_sgvpscC`st||tjƒS(s:Swap all the inputs of sgv0 and sgv1 (see reroute_inputs).(R[R!R"(RSRT((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/reroute.pyRˆscC`st||tjƒS(s=Re-route all the inputs of sgv0 to sgv1 (see reroute_inputs).(R[R!R&(RSRT((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/reroute.pyR scC`st||tjƒS(s=Swap all the outputs of sgv0 and sgv1 (see _reroute_outputs).(R]R!R"(RSRT((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/reroute.pyR ’scC`st||tjƒS(s@Re-route all the outputs of sgv0 to sgv1 (see _reroute_outputs).(R]R!R&(RSRT((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/reroute.pyR —scC`st||tjƒS(s;Swap the inputs and outputs of sgv1 to sgv0 (see _reroute).(R^R!R"(RSRT((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/reroute.pyR œscC`st||tjƒS(s?Re-route the inputs and outputs of sgv0 to sgv1 (see _reroute).(R^R!R&(RSRT((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/reroute.pyR ¡scC`s¹t|tjƒs*tdt|ƒƒ‚ntj|dtƒ}x>|D]6}||jkrFt dj |j |j ƒƒ‚qFqFWg|j D]}||krŠ|^qŠ|_ |j ƒdS(sRemove the control inputs cops from co. Warning: this function is directly manipulating the internals of the `tf.Graph`. Args: op: a `tf.Operation` from which to remove the control inputs. cops: an object convertible to a list of `tf.Operation`. Raises: TypeError: if op is not a `tf.Operation`. ValueError: if any cop in cops is not a control input of op. s Expected a tf.Operation, got: {}t allow_graphs{} is not a control_input of {}N(RLttf_opst OperationRNRORR=R%tcontrol_inputsRRtnamet_control_inputst_recompute_node_def(toptcopstcop((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/reroute.pyR¦s  +cC`st|tjƒs*tdt|ƒƒ‚ntj|dtƒ}x>|D]6}||jkrFt dj |j |j ƒƒ‚qFqFW|j |7_ |j ƒdS(suAdd the control inputs cops to co. Warning: this function is directly manipulating the internals of the tf.Graph. Args: op: a tf.Operation to which the control inputs are added. cops: an object convertible to a list of `tf.Operation`. Raises: TypeError: if op is not a tf.Operation ValueError: if any cop in cops is already a control input of op. s Expected a tf.Operation, got: {}R_s#{} is already a control_input of {}N(RLR`RaRNRORR=R%RbRRRcRdRe(RfRgRh((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/reroute.pyRÀs  N( R,t __future__RRRttensorflow.contrib.graph_editorRRttensorflow.python.frameworkRR`t__all__R tobjectR!R/R;RCRRRWR[R]R^RR R R R R RR(((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/reroute.pytsD   "I 7