o ?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)Z tensor_name componentsZ output_indexrrr_ParseTensorName's          rccsF|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) isinstancerZGraphDefZ MergeFrom TypeErrornoder getop_SetDefaultAttrValues) graph_defZ 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!rZbytes_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  r8cCs@|durdStdd|Dstd|dtdd|DS)z9Type-checks and possibly canonicalizes `return_elements`.Ncsr*r+r,r-xrrrr/r0z._ProcessReturnElementsParam..z?Argument `return_elements` must be a list of strings. Obtained rcss|]}t|VqdSr+ras_strr9rrrr/s)r4r"tuple)return_elementsrrr_ProcessReturnElementsParams r?cCs"|jD] }||jkr|SqdSr+)attrname)Z attr_namer(attr_defrrr_FindAttrInOpDefs  rCcCsdd|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)rA)r-r%rrr sz'_RemoveDefaultAttrs..N default_value) r%r#r r$r6r@rCHasFieldrE)producer_op_listr'Zproducer_op_dictr#r(Zproducer_op_defkeyrBrrr_RemoveDefaultAttrss      rIcCshtdd|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. csr*r+)r!r ZTensor)r-vrrrr/sz)_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().Z_inputscSsi|] \}}|t|qSr)r Zconvert_to_tensor)r-r.rJrrrrDsz*_ConvertInputMapValues..N)r4valuesrr name_scopeitems)rAr7rrr_ConvertInputMapValuess  rOFcCst||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_apiZ!TF_ImportGraphDefOptionsSetPrefixZ(TF_ImportGraphDefOptionsSetUniquifyNamesZ.TF_ImportGraphDefOptionsSetPropagateDeviceSpecrNrr<rZ _as_tf_outputoperZ.TF_ImportGraphDefOptionsRemapControlDependencyrZ'TF_ImportGraphDefOptionsAddInputMappingZ'TF_ImportGraphDefOptionsAddReturnOutputZ*TF_ImportGraphDefOptionsAddReturnOperationZ8TF_ImportGraphDefOptionsSetValidateColocationConstraints)optionsprefixr7r>validate_colocation_constraintspropagate_device_specZ input_srcZ input_dstZsrc_nameZdst_opZsrc_idxZ dst_outputrAZop_nameindexrrr _PopulateTFImportGraphDefOptionss<           rWc Csi}|jddD].}|j}|dt|}|r|||<qt| ||Wdn1s1wYq|D]C\}}d}|D]3}z||} Wntyit s\t |r^YqCt d|d|jw| jrvtj| j}qwqC|r~||q;dS)z3Processes the newly-added TF_Operations in `graph`.F)Zcompute_devicesrKNzSpecified colocation to an op: z+ that does not exist during import for op: )Z_add_new_tf_operationsrZ _set_device_GetColocationNamesr Z_apply_device_functionsrNZ_get_operation_by_nameKeyErrorrenabledr ZEnableControlFlowV2rrApydevZ DeviceSpecZ from_string) graphZcolocation_pairsZnew_opZoriginal_devicecolocation_namesr%Z coloc_op_listZ coloc_deviceZ coloc_op_nameZcoloc_oprrr_ProcessNewOpss>        r^cCsng}z|d}Wn tyYdSw|D]}t|}|dr4|tdd}||jkr4||q|S)zSet any default attr values in `node_def` that aren't present.rENvalue)r%rAr@rFZ WhichOneofZCopyFromrE)Znode_defr(rBrHrdrrrr&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). )r7r>rArG_import_graph_def_internal)r'r7r>rArfrGrrrrebs4cCst|d||dS)zCLike import_graph_def but does not validate colocation constraints.F)rTrArUrg)r'rArUrrrimport_graph_def_for_functions riTc 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/rKcSsg|]}t|qSrr;)r-srrr s z._import_graph_def_internal..z, z;Attempted to map inputs that were not found in graph_def: [])#r)r8r?rIr Zget_default_graphrMrLendswithrOrZScopedTFImportGraphDefOptionsrRrWZ_mutation_lockZ tf_bufferZSerializeToStringZ_c_graphr$rPZ!TF_GraphImportGraphDefWithResultsZScopedTFImportGraphDefResultsrZInvalidArgumentErrorrstrr^Zlibraryr Z from_libraryZ add_to_graphZ:TF_ImportGraphDefResultsMissingUnusedInputMappings_wrapperrbjoinrc)r'r7r>rTrArGrUr\scoperSZscoped_optionsrRZ serializedZc_graphrbeZ functionsfZmissing_unused_input_keysZ missing_keysrrrrhsr4           rh)F)NNNNN)NF)NNTNNF),__doc__ contextlibZtensorflow.core.frameworkrZtensorflow.pythonrZtensorflow.python.clientrrPZtensorflow.python.frameworkrrr[rr r r r Ztensorflow.python.opsr Ztensorflow.python.utilrZ"tensorflow.python.util.deprecationrZ tensorflow.python.util.tf_exportrrrcontextmanagerr r)r8r?rCrIrOrWr^rXrcr&rerirhrrrrsf              &   ! $4 :