o ?e8 @svdZddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z e ddd gd dd d Z dS)z-Utility functions for reading/writing graphs.N) text_format)byte_swap_tensor)ops)file_io) tf_exportzio.write_graphztrain.write_graph)v1TcCst|tjr |}n|}tjdkr&t|drt|ddnt |dd| ds0t |t j||}|rFt |tj|dd|St ||jdd|S) aWrites a graph proto to a file. The graph is written as a text proto unless `as_text` is `False`. ```python v = tf.Variable(0, name='my_variable') sess = tf.compat.v1.Session() tf.io.write_graph(sess.graph_def, '/tmp/my-model', 'train.pbtxt') ``` or ```python v = tf.Variable(0, name='my_variable') sess = tf.compat.v1.Session() tf.io.write_graph(sess.graph, '/tmp/my-model', 'train.pbtxt') ``` Args: graph_or_graph_def: A `Graph` or a `GraphDef` protocol buffer. logdir: Directory where to write the graph. This can refer to remote filesystems, such as Google Cloud Storage (GCS). name: Filename for the graph. as_text: If `True`, writes the graph as an ASCII proto. Returns: The path of the output proto file. bignodelittlezgs:)Z float_formatT)Z deterministic) isinstancerZGraphZ as_graph_defsys byteorderhasattrrZ!swap_tensor_content_in_graph_nodeZ%swap_tensor_content_in_graph_function startswithrZrecursive_create_dirospathjoinZatomic_write_string_to_filerZMessageToStringZSerializeToString)Zgraph_or_graph_defZlogdirnameZas_textZ graph_defrre/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/framework/graph_io.py write_graphs2       r)T)__doc__rZos.pathr Zgoogle.protobufrZtensorflow.python.frameworkrrZtensorflow.python.lib.iorZ tensorflow.python.util.tf_exportrrrrrrs