o i eaZ@s~dZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl m Z dd l mZdd l mZdd l mZdd l mZdd lmZddlmZddlmZddlmZddZddZejddZddZddZddZ ddZ!dd Z"d!d"Z# #d8d$d%Z$d&d'Z%d(d)Z&d*d+Z'd,d-Z(ed.d/edd0d1     d9d2d/Z) #d:d3d4Z*   5   #d;d6d7Z+dS) ("foo", 0) * "foo:7" -> ("foo", 7) * "foo" -> ("foo", 0) * "foo:bar:baz" -> ValueError Args: tensor_name: The name of a tensor. Returns: A tuple containing the operation name, and the output index. Raises: ValueError: If `tensor_name' cannot be interpreted as the name of a tensor. :zCannot convert zX to a tensor name. Second component of the name following the `:` should be an int. Got .rzCannot convert 'zN' to a tensor name. Tensor names should not contain more than 1 `:`. Obtained )splitlenint ValueError) tensor_name components output_indexrrr_ParseTensorName's          r"ccsF|rt| dVWddS1swYdSdVdS)z=Applies the given device only if device is not None or empty.N)r rrrrr _MaybeDeviceMs  " r#cCspt|tjs z|}t}||W|Stytdw|jD]}t|j}|dur0q#t ||q#|S)z3Type-checks and possibly canonicalizes `graph_def`.z.Argument `graph_def` must be a GraphDef proto.N) isinstancerGraphDef MergeFrom TypeErrornoder getop_SetDefaultAttrValues) graph_def old_graph_defr(op_defrrr_ProcessGraphDefParamWs     r/cCs^|duri}|St|tstdt|jtdd|Ds-tdt||S)z3Type-checks and possibly canonicalizes `input_map`.Nz4Argument `input_map` must be a dictionary. Obtained cs|] }t|tjVqdSNr$rbytes_or_text_types).0krrr }  z(_ProcessInputMapParam..zBAll keys for argument `input_map` must be strings. Obtained keys: )r$dictr'type__name__allkeyslist) input_maprrr_ProcessInputMapParamus  r?cCs@|durdStdd|Dstd|dtdd|DS)z9Type-checks and possibly canonicalizes `return_elements`.Ncsr0r1r2r4xrrrr6r7z._ProcessReturnElementsParam..z?Argument `return_elements` must be a list of strings. Obtained rcss|]}t|VqdSr1ras_strr@rrrr6s)r;r'tuple)return_elementsrrr_ProcessReturnElementsParams rFcCs"|jD] }||jkr|SqdSr1)attrname) attr_namer.attr_defrrr_FindAttrInOpDefs  rKcCsdd|jD}|jD]>}|j|vrIt|j}|durq ||j}t|jD]!}t||durHt||}|rH|drH|j||jkrH|j|=q'q dS)a Removes unknown default attrs according to `producer_op_list`. Removes any unknown attrs in `graph_def` (i.e. attrs that do not appear in registered OpDefs) that have a default value in `producer_op_list`. Args: producer_op_list: OpList proto. graph_def: GraphDef proto cSsi|]}|j|qSr)rH)r4r*rrr sz'_RemoveDefaultAttrs..N default_value) r*r(r r)r=rGrKHasFieldrM)producer_op_listr,producer_op_dictr(r.producer_op_defkeyrJrrr_RemoveDefaultAttrss     rScCshtdd|Ds2|dkrtdtddd|D}Wd|S1s-wY|S) aEnsures all input map values are tensors. This should be called from inside the import name scope. Args: name: the `name` argument passed to import_graph_def input_map: the `input_map` argument passed to import_graph_def. Returns: An possibly-updated version of `input_map`. Raises: ValueError: if input map values cannot be converted due to empty name scope. csr0r1)r$r Tensor)r4vrrrr6sz)_ConvertInputMapValues..ztf.import_graph_def() requires a non-empty `name` if `input_map` contains non-Tensor values. Try calling tf.convert_to_tensor() on `input_map` values before calling tf.import_graph_def()._inputscSsi|] \}}|t|qSr)r convert_to_tensor)r4r5rUrrrrLsz*_ConvertInputMapValues..N)r;valuesrr name_scopeitems)rHr>rrr_ConvertInputMapValuess  r\FcCst||t|dt|||D];\}}t|}|dr:t|dd}|j } t ||| qt |\}} t|}|} t ||| | q|pUgD]"} d| vrot | \} }t| } t || |qVt|t| qVt||dS)z1Populates the TF_ImportGraphDefOptions `options`.TrrNr)c_api!TF_ImportGraphDefOptionsSetPrefix(TF_ImportGraphDefOptionsSetUniquifyNames.TF_ImportGraphDefOptionsSetPropagateDeviceSpecr[rrCr _as_tf_outputoper.TF_ImportGraphDefOptionsRemapControlDependencyr"'TF_ImportGraphDefOptionsAddInputMapping'TF_ImportGraphDefOptionsAddReturnOutput*TF_ImportGraphDefOptionsAddReturnOperation8TF_ImportGraphDefOptionsSetValidateColocationConstraints)optionsprefixr>rEvalidate_colocation_constraintspropagate_device_spec input_src input_dstsrc_namedst_opsrc_idx dst_outputrHop_nameindexrrr _PopulateTFImportGraphDefOptionss<           rtc Csi}|jddD].}|j}|dt|}|r|||<qt| ||Wdn1s1wYq|D]C\}}d}|D]3}z||} Wntyit s\t |r^YqCt d|d|jw| jrvtj| j}nqC|r~||q;dS)z3Processes the newly-added TF_Operations in `graph`.F)compute_devicesrVNzSpecified colocation to an op: z+ that does not exist during import for op: )_add_new_tf_operationsr _set_device_GetColocationNamesr#_apply_device_functionsr[_get_operation_by_nameKeyErrorrenabledr EnableControlFlowV2rrHpydev DeviceSpec from_string) graphcolocation_pairsnew_oporiginal_devicecolocation_namesr* coloc_op_list coloc_device coloc_op_namecoloc_oprrr_ProcessNewOpss>        rcCsng}z|d}Wn tyYdSw|D]}t|}|dr4|tdd}||jkr4||q|S)zSet any default attr values in `node_def` that aren't present.rMNvalue)r*rHrGrN WhichOneofCopyFromrM)node_defr.rJrRrrrrr+Ws   r+zgraph_util.import_graph_defimport_graph_defzfPlease file an issue at https://github.com/tensorflow/tensorflow/issues if you depend on this feature.op_dictcCs~t|||||dS)aImports the graph from `graph_def` into the current default `Graph`. This function provides a way to import a serialized TensorFlow [`GraphDef`](https://www.tensorflow.org/code/tensorflow/core/framework/graph.proto) protocol buffer, and extract individual objects in the `GraphDef` as `tf.Tensor` and `tf.Operation` objects. Once extracted, these objects are placed into the current default `Graph`. See `tf.Graph.as_graph_def` for a way to create a `GraphDef` proto. Args: graph_def: A `GraphDef` proto containing operations to be imported into the default graph. input_map: A dictionary mapping input names (as strings) in `graph_def` to `Tensor` objects. The values of the named input tensors in the imported graph will be re-mapped to the respective `Tensor` values. return_elements: A list of strings containing operation names in `graph_def` that will be returned as `Operation` objects; and/or tensor names in `graph_def` that will be returned as `Tensor` objects. name: (Optional.) A prefix that will be prepended to the names in `graph_def`. Note that this does not apply to imported function names. Defaults to `"import"`. op_dict: (Optional.) Deprecated, do not use. producer_op_list: (Optional.) An `OpList` proto with the (possibly stripped) list of `OpDef`s used by the producer of the graph. If provided, unrecognized attrs for ops in `graph_def` that have their default value according to `producer_op_list` will be removed. This will allow some more `GraphDef`s produced by later binaries to be accepted by earlier binaries. Returns: A list of `Operation` and/or `Tensor` objects from the imported graph, corresponding to the names in `return_elements`, and None if `returns_elements` is None. Raises: TypeError: If `graph_def` is not a `GraphDef` proto, `input_map` is not a dictionary mapping strings to `Tensor` objects, or `return_elements` is not a list of strings. ValueError: If `input_map`, or `return_elements` contains names that do not appear in `graph_def`, or `graph_def` is not well-formed (e.g. it refers to an unknown tensor). )r>rErHrO_import_graph_def_internal)r,r>rErHrrOrrrrbs4cCst|d||dS)zCLike import_graph_def but does not validate colocation constraints.F)rjrHrkr)r,rHrkrrrimport_graph_def_for_functions rTc Cst|}t|}t|}|durt||t}t|d|}|r3|ds,J|dd} nd} t ||}Wdn1sDwYt } | j } t | | |||||Zt |?} z#|j} t| | | }Wdn1s~wYt |}Wntjy}ztt|d}~wwWdn1swYt|Wdn1swY|jr|jjrt|j}|D]}||qt|j }|rdd|D}d!|}td |d |durdSt"|||j S) ag Imports the graph from `graph_def` into the current default `Graph`. This function provides a way to import a serialized TensorFlow [`GraphDef`](https://www.tensorflow.org/code/tensorflow/core/framework/graph.proto) protocol buffer, and extract individual objects in the `GraphDef` as `tf.Tensor` and `tf.Operation` objects. Once extracted, these objects are placed into the current default `Graph`. See `tf.Graph.as_graph_def` for a way to create a `GraphDef` proto. Args: graph_def: A `GraphDef` proto containing operations to be imported into the default graph. input_map: A dictionary mapping input names (as strings) in `graph_def` to `Tensor` objects. The values of the named input tensors in the imported graph will be re-mapped to the respective `Tensor` values. return_elements: A list of strings containing operation names in `graph_def` that will be returned as `Operation` objects; and/or tensor names in `graph_def` that will be returned as `Tensor` objects. validate_colocation_constraints: Whether to validate colocation constraints. name: (Optional.) A prefix that will be prepended to the names in `graph_def`. Note that this does not apply to imported function names. Defaults to `"import"`. producer_op_list: (Optional.) An `OpList` proto with the (possibly stripped) list of `OpDef`s used by the producer of the graph. If provided, unrecognized attrs for ops in `graph_def` that have their default value according to `producer_op_list` will be removed. This will allow some more `GraphDef`s produced by later binaries to be accepted by earlier binaries. propagate_device_spec: Whether to propagate assigned device information when importing a graph from a GraphDef into the current default `Graph`. Returns: A list of `Operation` and/or `Tensor` objects from the imported graph, corresponding to the names in `return_elements`, and None if `returns_elements` is None. Raises: TypeError: If `graph_def` is not a `GraphDef` proto, `input_map` is not a dictionary mapping strings to `Tensor` objects, or `return_elements` is not a list of strings. ValueError: If `input_map`, or `return_elements` contains names that do not appear in `graph_def`, or `graph_def` is not well-formed (e.g. it refers to an unknown tensor). Nimport/rVcSsg|]}t|qSrrB)r4srrr s z._import_graph_def_internal..z, z;Attempted to map inputs that were not found in graph_def: [])#r/r?rFrSr get_default_graphrZrYendswithr\rScopedTFImportGraphDefOptionsrhrt_mutation_lock tf_bufferSerializeToString_c_graphr)r]!TF_GraphImportGraphDefWithResultsScopedTFImportGraphDefResultsrInvalidArgumentErrorrstrrlibraryr from_library add_to_graph:TF_ImportGraphDefResultsMissingUnusedInputMappings_wrapperrjoinr)r,r>rErjrHrOrkrscoperiscoped_optionsrh serializedc_graphre functionsfmissing_unused_input_keys missing_keysrrrrsr4           r)F)NNNNN)NF)NNTNNF),__doc__ contextlibtensorflow.core.frameworkrtensorflow.pythonrtensorflow.python.clientrr]tensorflow.python.frameworkrrr~rr r r r tensorflow.python.opsr tensorflow.python.utilr"tensorflow.python.util.deprecationr tensorflow.python.util.tf_exportrrr"contextmanagerr#r/r?rFrKrSr\rtrrxrr+rrrrrrrsf              &   ! $4 :