| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| make_test_graphs.py | 171 def write_graph(build_graph, out_dir): function
182 write_graph(tfadd, FLAGS.out_dir)
183 write_graph(tfadd_with_ckpt, FLAGS.out_dir)
184 write_graph(tfadd_with_ckpt_saver, FLAGS.out_dir)
185 write_graph(tfassert_eq, FLAGS.out_dir)
186 write_graph(tfcond, FLAGS.out_dir)
187 write_graph(tffunction, FLAGS.out_dir)
188 write_graph(tfgather, FLAGS.out_dir)
189 write_graph(tfmatmul, FLAGS.out_dir)
190 write_graph(tfmatmulandadd, FLAGS.out_dir
[all...] |
| /external/tensorflow/tensorflow/contrib/receptive_field/python/util/examples/ |
| write_inception_resnet_v2_graph.py | 43 graph_io.write_graph(g.as_graph_def(), cmd_args.graph_dir,
|
| /external/tensorflow/tensorflow/python/framework/ |
| graph_io.py | 30 @tf_export('io.write_graph', v1=['io.write_graph', 'train.write_graph'])
31 def write_graph(graph_or_graph_def, logdir, name, as_text=True): function
39 tf.train.write_graph(sess.graph_def, '/tmp/my-model', 'train.pbtxt')
47 tf.train.write_graph(sess.graph, '/tmp/my-model', 'train.pbtxt')
|
| meta_graph.py | [all...] |
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| strip_pruning_vars.py | 71 graph_io.write_graph(final_graph_def, output_dir, filename, as_text=False)
|
| /external/tensorflow/tensorflow/python/tools/ |
| optimize_for_inference.py | 98 graph_io.write_graph(output_graph_def,
|
| strip_unused_test.py | 55 graph_io.write_graph(sess.graph, self.get_temp_dir(), input_graph_name)
118 graph_io.write_graph(sess.graph, self.get_temp_dir(), input_graph_name)
|
| freeze_graph_test.py | 74 graph_io.write_graph(sess.graph, self.get_temp_dir(), input_graph_name)
305 graph_io.write_graph(sess.graph, self.get_temp_dir(), input_graph_name)
|
| /external/tensorflow/tensorflow/lite/testing/model_coverage/ |
| model_coverage_lib_test.py | 36 from tensorflow.python.training.training_util import write_graph
43 write_graph(sess.graph_def, '', graph_def_file, False)
|
| /external/tensorflow/tensorflow/python/training/ |
| training_util.py | 39 write_graph = graph_io.write_graph variable
|
| training.py | 102 from tensorflow.python.training.training_util import write_graph
|
| basic_session_run_hooks.py | 556 training_util.write_graph(
[all...] |
| supervisor.py | 633 training_util.write_graph(self._graph.as_graph_def(add_shapes=True),
[all...] |
| saver_test.py | [all...] |
| /external/tensorflow/tensorflow/python/keras/ |
| callbacks_v1.py | 71 write_graph: whether to visualize the graph in TensorBoard. The log file
72 can become quite large when write_graph is set to True.
121 write_graph=True,
139 self.write_graph = write_graph
169 if not model.run_eagerly and self.write_graph:
172 elif self.write_graph:
|
| callbacks.py | [all...] |
| callbacks_test.py | [all...] |
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| freeze.py | 171 tf.train.write_graph(
|
| train.py | 194 tf.train.write_graph(sess.graph_def, FLAGS.train_dir,
|
| /external/tensorflow/tensorflow/python/tpu/ |
| async_checkpoint.py | 104 training_util.write_graph(
137 logging.info("Waiting for any pending write_graph to finish.")
|
| tensor_tracer.py | [all...] |
| /external/tensorflow/tensorflow/lite/python/ |
| lite_test.py | 42 from tensorflow.python.training.training_util import write_graph
641 write_graph(sess.graph_def, '', graph_def_file, False)
676 write_graph(sess.graph_def, '', graph_def_file, False)
704 write_graph(sess.graph_def, '', graph_def_file, False)
722 write_graph(sess.graph_def, '', graph_def_file, True)
[all...] |
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| tf_trt_integration_test_base.py | 363 graph_io.write_graph(gdef, temp_dir, graph_name)
|
