HomeSort by relevance Sort by last modified time
    Searched refs:main_op (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/python/saved_model/
main_op_impl.py 28 @tf_export('saved_model.main_op.main_op')
29 def main_op(): function
45 @tf_export('saved_model.main_op.main_op_with_restore')
59 with ops.control_dependencies([main_op()]):
main_op.py 26 from tensorflow.python.saved_model.main_op_impl import main_op
32 "main_op",
builder_impl.py 157 def _add_main_op(self, main_op):
161 main_op: Main op to run as part of graph initialization.
166 if main_op is not None:
167 if not isinstance(main_op, ops.Operation):
168 raise TypeError("main_op needs to be an Operation: %r" % main_op)
169 ops.add_to_collection(constants.MAIN_OP_KEY, main_op)
244 main_op=None,
264 main_op: Op or group of ops to execute when the graph is loaded. Note
265 that when the main_op is specified it is run after the restore op a
    [all...]
saved_model.py 27 from tensorflow.python.saved_model import main_op
44 "main_op",
saved_model_test.py 43 from tensorflow.python.saved_model import main_op
607 # Set up an assignment op to be run as part of the main_op.
608 with ops.control_dependencies([main_op.main_op()]):
614 sess, ["foo"], main_op=custom_main_op)
624 # the main_op, following a restore.
    [all...]
  /external/tensorflow/tensorflow/examples/saved_model/
saved_model_half_plus_two.py 207 main_op=tf.group(tf.saved_model.main_op.main_op(),
  /external/tensorflow/tensorflow/contrib/meta_graph_transform/
meta_graph_transform_test.py 88 main_op = 'main_op'
93 saved_model_constants.MAIN_OP_KEY: [main_op],
98 meta_graph_transform._get_shared_init_op(main_and_legacy), main_op)
128 main_op = 'main_op'
133 saved_model_constants.MAIN_OP_KEY: [main_op],
149 saved_model_constants.MAIN_OP_KEY: [main_op],
154 table_init_names, main_op, saver_def,

Completed in 232 milliseconds