ó V¾÷Xc@`s^dZddlmZddlmZddlmZddlZddlZddlZddlm Z ddl m Z ddl m Z dd l m Z dd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZd„Zd„Zddd„Z d„Z!de"fd„ƒYZ#dS(s0Export a TensorFlow model. See: go/tf-exporter i(tabsolute_import(tdivision(tprint_functionN(tAny(t constants(tgc(t manifest_pb2(t graph_pb2(tdtypes(tops(tgfile(t tf_logging(tsaver(t training_util(tcompatcC`s·tjd|ƒtj|ƒx“|jƒD]…\}}tjjtj |ƒtj |ƒƒ}tjd||ƒtj |ƒrŸtjd|ƒtj |ƒntj ||ƒq*WdS(søCallback to copy files using `gfile.Copy` to an export directory. This method is used as the default `assets_callback` in `Exporter.init` to copy assets from the `assets_collection`. It can also be invoked directly to copy additional supplementary files into the export directory (in which case it is not a callback). Args: files_to_copy: A dictionary that maps original file paths to desired basename in the export directory. export_dir_path: Directory to copy the files to. s'Write assets into: %s using gfile_copy.sCopying asset %s to path %s.sRemoving file %s.N( tloggingtinfoR tMakeDirstitemstostpathtjoinRtas_bytestExiststRemovetCopy(t files_to_copytexport_dir_pathtsource_filepathtbasenametnew_path((sl/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/session_bundle/exporter.pytgfile_copy_callback+s   cC`s4tjƒ}|j|jj_|j|jj_|S(sÆCreates a regression signature. Args: input_tensor: Tensor specifying the input to a graph. output_tensor: Tensor specifying the output of a graph. Returns: A Signature message. (Rt Signaturetnametregression_signaturetinputt tensor_nametoutput(t input_tensort output_tensort signature((sl/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/session_bundle/exporter.pyR"Hs cC`sdtjƒ}|j|jj_|dk r?|j|jj_n|dk r`|j|jj_n|S(sCreates a classification signature. Args: input_tensor: Tensor specifying the input to a graph. classes_tensor: Tensor specifying the output classes of a graph. scores_tensor: Tensor specifying the scores of the output classes. Returns: A Signature message. N( RR R!tclassification_signatureR#R$tNonetclassestscores(R&tclasses_tensort scores_tensorR(((sl/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/session_bundle/exporter.pyR)Xs   cC`sFtjƒ}x3tj|ƒD]"\}}|j|jj|_qW|S(sšCreates a generic signature of name to Tensor name. Args: name_tensor_map: Map from logical name to Tensor. Returns: A Signature message. (RR tsixt iteritemsR!tgeneric_signaturetmapR$(tname_tensor_mapR(R!ttensor((sl/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/session_bundle/exporter.pyR1ns tExportercB`sMeZdZd„Zddeddded„Zddd„Zd„Z RS(s[Exporter helps package a TensorFlow model for serving. Args: saver: Saver object. cC`sD|jƒ}|jdƒtjd|ƒ|_t|_i|_dS(Nt max_to_keept saver_def(t as_saver_deft ClearFieldttf_savertSavert_savertFalset _has_initt_assets_to_copy(tselfR R7((sl/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/session_bundle/exporter.pyt__init__„s    cC`s}|dkri}ng}|rxk|D]`} |j| ƒ} | sVtd| ƒ‚ntjj| ƒ} |j| | fƒ| |j| t RuntimeErrortTrueRtGraphDeftCopyFromR tget_default_grapht as_graph_deftnodetdeviceRtPacktadd_to_collectionRt GRAPH_KEYt isinstancet Operationt TypeErrort INIT_OP_KEYRt Signaturestdefault_signatureR/R0tnamed_signaturestSIGNATURES_KEYt AssetFiletfilenameR!ttensor_bindingR$t ASSETS_KEYt_assets_callback(R@t graph_deftinit_opt clear_devicestdefault_graph_signaturetnamed_graph_signaturestassets_collectiontassets_callbacktassetst asset_tensortasset_filepathRtcopyRLt graph_any_buftsignatures_prototsignature_nameR(tsignatures_any_bufRZR4tassett asset_any_buf((sl/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/session_bundle/exporter.pytinitsZ                c `sÆ|jstdƒ‚nˆjdƒr4ˆd ‰ntj||ƒ}tjjtj ˆƒtj t j |ƒƒ}t j |ƒr–td|ƒ‚ntj|ƒd}t j|ƒ|jj|tjjtj|ƒtjt jƒƒdt jƒ|jrU|jrUtjjtj |ƒtj t jƒƒ}t j|ƒ|j|j|ƒnt j||ƒ|r‡fd†} tj|ƒ} x6| tjˆd| ƒƒD]} t j| jƒq¥Wn|S( s“Exports the model. Args: export_dir_base: A string path to the base export dir. global_step_tensor: An Tensor or tensor name providing the global step counter to append to the export directory path and set in the manifest version. sess: A Session to use to save the parameters. exports_to_keep: a gc.Path filter function used to determine the set of exports to keep. If set to None, all versions will be kept. Returns: The string path to the exported directory. Raises: RuntimeError: if init is not called. RuntimeError: if the export would overwrite an existing directory. sinit must be called firstt/iÿÿÿÿsVOverwriting exports can cause corruption and are not allowed. Duplicate export dir: %ss-tmptmeta_graph_suffixc`sFtjdˆd|jƒ}|s'dS|jdt|jdƒƒƒS(Nt^s /(\d{8})$texport_versioni(tretmatchRR*t_replacetinttgroup(RRu(texport_dir_base(sl/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/session_bundle/exporter.pytparser"sRz(R>RFtendswithR t global_stepRRRRRRtVERSION_FORMAT_SPECIFIERR Rtas_textRR<tsavetEXPORT_BASE_NAMEtEXPORT_SUFFIX_NAMER]R?tASSETS_DIRECTORYtRenameRtnegationt get_pathstDeleteRecursively( R@Rytglobal_step_tensortsesstexports_to_keepR|t export_dirttmp_export_dirt assets_dirRztpaths_to_deletetp((Rysl/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/session_bundle/exporter.pytexportàs>             "cC`s¡t|tjƒs!tdƒ‚n|jjdkrBtdƒ‚n|jtjkrctdƒ‚n|jj dƒj }t |ƒdkr™tdƒ‚n|dS( s<Returns the filepath value stored in constant `path_tensor`.stensor is not a TensortConsts#Only constants tensor are supportedsFile paths should be stringtvalueis!Only scalar tensors are supportedi( RQR tTensorRStopttypetdtypeRtstringtget_attrt string_valtlen(R@t path_tensort str_value((sl/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/session_bundle/exporter.pyRC.sN( t__name__t __module__t__doc__RAR*R=RRoRRC(((sl/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/session_bundle/exporter.pyR5}s  M J($Ržt __future__RRRRRtR/tgoogle.protobuf.any_pb2Rt!tensorflow.contrib.session_bundleRRRttensorflow.core.frameworkRttensorflow.python.frameworkRR ttensorflow.python.platformR R Rttensorflow.python.trainingR R:R ttensorflow.python.utilRRR"R*R)R1tobjectR5(((sl/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/session_bundle/exporter.pyts0