ó V¾÷Xc@`södZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z dd l m Z d d d d dddgZ d„Zd„Zed„Zdd„Zeddd„Zed„Zd„ZdS(s#Various function for graph editing.i(tabsolute_import(tdivision(tprint_function(treroute(tselect(tsubgraph(tutil(t array_opstdetach_control_inputstdetach_control_outputst detach_inputstdetach_outputstdetachtconnecttbypasscC`sbtj|ƒ}xL|jD]A}g|jD]}||jkr)|^q)}tj||ƒqWdS(sÚDetach all the external control inputs of the subgraph sgv. Args: sgv: the subgraph view to be detached. This argument is converted to a subgraph using the same rules as the function subgraph.make_view. N(Rt make_viewtopstcontrol_inputsRtremove_control_inputs(tsgvtoptcoptcops((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/edit.pyR&s+cC`s—t|tjƒs*tdt|ƒƒ‚n|jƒtj|ƒ}xM|jD]B}x9|j |ƒD](}||jkrct j ||ƒqcqcWqMWdS(sDetach all the external control outputs of the subgraph sgv. Args: sgv: the subgraph view to be detached. This argument is converted to a subgraph using the same rules as the function subgraph.make_view. control_outputs: a util.ControlOutputs instance. s'Expected a util.ControlOutputs, got: {}N( t isinstanceRtControlOutputst TypeErrorttypetupdateRRRtgetRR(Rtcontrol_outputsRR((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/edit.pyR 3s c C`sŒtj|ƒ}|jjƒ?g|jD]*}tjd|jdtj |ƒƒ^q)}WdQXt j ||ƒ|r‚t |ƒn||fS(suDetach the inputs of a subgraph view. Args: sgv: the subgraph view to be detached. This argument is converted to a subgraph using the same rules as the function subgraph.make_view. Note that sgv is modified in place. control_inputs: if True control_inputs are also detached. Returns: A tuple `(sgv, input_placeholders)` where `sgv` is a new subgraph view of the detached subgraph; `input_placeholders` is a list of the created input placeholders. Raises: StandardError: if sgv cannot be converted to a SubGraphView using the same rules than the function subgraph.make_view. tdtypetnameN( RRtgrapht as_defaulttinputst tf_array_opst placeholderRRtplaceholder_nameRt swap_inputsR(RRtinput_ttinput_placeholders((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/edit.pyR Fs= c C`stj|ƒ}|jgt|jƒD]\}}|jƒr%|^q%ƒ}tj|jƒƒ}|jgt|jƒD]!\}}||jkrw|^qwƒ}|j j ƒ*g|jD]}t j |ƒ^q»}WdQXt j||ƒ|dk rt||ƒn||fS(s°Detach the output of a subgraph view. Args: sgv: the subgraph view to be detached. This argument is converted to a subgraph using the same rules as the function subgraph.make_view. Note that sgv is modified in place. control_outputs: a util.ControlOutputs instance or None. If not None the control outputs are also detached. Returns: A tuple `(sgv, output_placeholders)` where `sgv` is a new subgraph view of the detached subgraph; `output_placeholders` is a list of the created output placeholders. Raises: StandardError: if sgv cannot be converted to a SubGraphView using the same rules than the function subgraph.make_view. N(RRt remap_outputst enumeratetoutputst consumerst SubGraphViewt remap_inputsR"R R!Rtmake_placeholder_from_tensorRt swap_outputstNoneR ( RRt output_idtoutput_ttsgv_t consumers_sgvtinput_idR'toutput_placeholders((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/edit.pyR es ( cC`sRtj|||ƒ\}}t||ƒ\}}t||ƒ\}}|||fS(sPDetach both the inputs and the outputs of a subgraph view. Args: sgv: the subgraph view to be detached. This argument is converted to a subgraph using the same rules as the function subgraph.make_view. Note that sgv is modified in place. control_inputs: A boolean indicating whether control inputs are enabled. control_outputs: An instance of util.ControlOutputs or None. If not None, control outputs are enabled. control_ios: An instance of util.ControlOutputs or None. If not None, both control inputs and control outputs are enabled. This is equivalent to set control_inputs to True and control_outputs to the util.ControlOutputs instance. Returns: A tuple `(sgv, detached_inputs, detached_outputs)` where: `sgv` is a new subgraph view of the detached subgraph; `detach_inputs` is a list of the created input placeholders; `detach_outputs` is a list of the created output placeholders. Raises: StandardError: if sgv cannot be converted to a SubGraphView using the same rules than the function subgraph.make_view. (Rt check_ciosR R (RRRt control_iost_tdetached_inputstdetached_outputs((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/edit.pyR s  cC`sptj|ƒ}tj|ƒ}tj||ƒ|rAt|ƒntjd|jƒ}tj||ƒ||fS(s Connect the outputs of sgv0 to the inputs of sgv1. Args: sgv0: the first subgraph to have its outputs swapped. This argument is converted to a subgraph using the same rules as the function subgraph.make_view. Note that sgv0 is modified in place. sgv1: the second subgraph to have its outputs swapped. This argument is converted to a subgraph using the same rules as the function subgraph.make_view. Note that sgv1 is modified in place. disconnect_first: if True the current outputs of sgv0 are disconnected. Returns: A tuple `(sgv0, sgv1)` of the now connected subgraphs. Raises: StandardError: if sgv0 or sgv1 cannot be converted to a SubGraphView using the same rules than the function subgraph.make_view. tpassthrough_ts( RRRt check_graphsR R-R+Rtreroute_inputs(tsgv0tsgv1tdisconnect_firstt sgv0_outputs((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/edit.pyR ¬s cC`sMtj|ƒ}t|jƒ}t|ƒ\}}tj||jƒ||fS(sQBypass the given subgraph by connecting its inputs to its outputs. Args: sgv: the subgraph view to be bypassed. This argument is converted to a subgraph using the same rules than the function subgraph.make_view. Note that sgv is modified in place. Returns: A tuple `(sgv, detached_inputs)` where: `sgv` is a new subgraph view of the bypassed subgraph; `detached_inputs` is a list of the created input placeholders. Raises: StandardError: if sgv cannot be converted to a SubGraphView using the same rules than the function subgraph.make_view. (RRtlistR"R Rt reroute_tsR+(Rt sgv_inputsR;((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/edit.pyRÉs N(t__doc__t __future__RRRttensorflow.contrib.graph_editorRRRRttensorflow.python.opsRR#t__all__RR tFalseR R1R R R R(((sf/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/graph_editor/edit.pyts,    (