Home
last modified time | relevance | path

Searched refs:init_op (Results 1 – 25 of 92) sorted by relevance

1234

/external/tensorflow/tensorflow/python/data/kernel_tests/
Ddataset_checkpoint_test.py69 init_op = iterator.initializer
73 return init_op, get_next, save_op, restore_op
80 init_op, get_next, save_op, _ = _build_graph(start, stop)
83 sess.run(init_op)
89 init_op, get_next, _, restore_op = _build_graph(start, stop)
91 sess.run(init_op)
100 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
103 sess.run(init_op)
118 init_op = iterator.initializer
122 return init_op, get_next, save_op, restore_op
[all …]
Dfrom_sparse_tensor_slices_test.py40 init_op = iterator.initializer
53 sess.run(init_op, feed_dict={st: sparse_feed})
71 sess.run(init_op, feed_dict={st: sparse_feed})
79 sess.run(init_op, feed_dict={st: sparse_feed})
Diterator_cluster_test.py139 init_op = iterator.initializer
144 sess.run(init_op)
172 init_op = iterator.initializer
176 sess.run(init_op)
/external/tensorflow/tensorflow/contrib/data/python/kernel_tests/
Dassert_element_shape_test.py54 init_op = iterator.initializer
57 sess.run(init_op)
98 init_op = iterator.initializer
101 sess.run(init_op)
127 init_op = iterator.initializer
130 sess.run(init_op)
153 init_op = iterator.initializer
156 sess.run(init_op)
197 init_op = iterator.initializer
200 sess.run(init_op)
[all …]
Dslide_dataset_op_test.py75 init_op = iterator.initializer
83 init_op,
139 init_op = iterator.initializer
147 init_op,
184 init_op = iterator.initializer
189 init_op,
212 init_op = iterator.initializer
216 sess.run(init_op)
241 init_op = iterator.initializer
245 sess.run(init_op)
[all …]
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Ddataset_serialization_test_base.py268 init_op, get_next_op = self._get_iterator_ops_from_collection(
273 self._initialize(init_op, sess)
426 init_op, get_next_op, saver = self._build_graph(
430 self._initialize(init_op, sess)
524 init_op, get_next_op = self._get_iterator_ops_from_collection(
527 init_op, get_next_op, saver = self._build_graph(
529 return init_op, get_next_op, saver
533 init_op, get_next_op, saver = get_ops()
538 self._initialize(init_op, sess)
541 self._initialize(init_op, sess)
[all …]
Drange_dataset_serialization_test.py61 init_op = iterator.initializer
65 return init_op, get_next, save_op, restore_op
72 init_op, get_next, save_op, _ = _build_graph(start, stop)
75 self.evaluate(init_op)
81 init_op, get_next, _, restore_op = _build_graph(start, stop)
83 self.evaluate(init_op)
92 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
95 self.evaluate(init_op)
/external/tensorflow/tensorflow/contrib/kafka/python/kernel_tests/
Dkafka_test.py55 init_op = iterator.make_initializer(repeat_dataset)
61 sess.run(init_op, feed_dict={topics: ["test:0:0:4"], num_epochs: 1})
68 sess.run(init_op, feed_dict={topics: ["test:0:5:-1"], num_epochs: 1})
76 init_op,
89 init_op,
/external/tensorflow/tensorflow/python/training/
Dsupervisor.py214 init_op=USE_DEFAULT, argument
318 self._init_init_op(init_op=init_op, init_feed_dict=init_feed_dict)
421 def _init_init_op(self, init_op=USE_DEFAULT, init_feed_dict=None): argument
430 if init_op is Supervisor.USE_DEFAULT:
431 init_op = self._get_first_op_from_collection(ops.GraphKeys.INIT_OP)
432 if init_op is None:
433 init_op = variables.global_variables_initializer()
434 ops.add_to_collection(ops.GraphKeys.INIT_OP, init_op)
435 self._init_op = init_op
536 def init_op(self): member in Supervisor
[all …]
Dsession_manager.py226 init_op=None, argument
292 if init_op is None and not init_fn and self._local_init_op is None:
295 if init_op is not None:
296 sess.run(init_op, feed_dict=init_feed_dict)
304 "Init op: %s, init fn: %s, error: %s" % (_maybe_name(init_op),
313 (_maybe_name(init_op), init_fn, self._local_init_op, msg))
Dsession_manager_test.py48 "", init_op=variables.global_variables_initializer())
59 init_op=variables.global_variables_initializer(),
90 init_op=variables.global_variables_initializer(),
113 init_op=None,
124 init_op=None,
490 sess = sm2.prepare_session("", init_op=v.initializer)
544 sess = sm2.prepare_session("", init_op=None)
586 sess = sm.prepare_session("", init_op=v.initializer)
608 sm2.prepare_session("", init_op=v.initializer)
625 sm2.prepare_session("", init_op=[v.initializer])
[all …]
Dqueue_runner_test.py258 init_op = variables.global_variables_initializer()
262 init_op.run()
275 init_op = variables.global_variables_initializer()
279 init_op.run()
288 init_op = variables.global_variables_initializer()
292 init_op.run()
305 init_op = variables.global_variables_initializer()
309 init_op.run()
/external/tensorflow/tensorflow/contrib/kinesis/python/kernel_tests/
Dkinesis_test.py65 init_op = iterator.make_initializer(repeat_dataset)
71 sess.run(init_op, feed_dict={stream: stream_name, num_epochs: 1})
109 init_op = iterator.make_initializer(repeat_dataset)
117 init_op, feed_dict={
126 init_op, feed_dict={
/external/tensorflow/tensorflow/python/saved_model/
Dloader_impl.py153 init_op = None
159 init_op = ops.get_collection(init_op_key)[0]
160 return init_op
399 init_op = get_init_op(meta_graph_def, import_scope)
400 if init_op is not None:
401 sess.run(fetches=[init_op], feed_dict=asset_tensors_dictionary)
Dload_v1_in_v2.py130 init_op = loader_impl.get_init_op(meta_graph_def)
132 if init_op is not None:
141 fetches=[wrapped.graph.as_graph_element(init_op)])
/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dlearning.py72 init_op=_USE_DEFAULT, argument
176 init_op=init_op,
/external/tensorflow/tensorflow/python/grappler/
Dmemory_optimizer_test.py131 init_op = variables.global_variables_initializer()
133 return (metagraph, init_op.name, train_op.name, loss.name)
234 init_op = graph.get_operation_by_name(init_op_name)
238 self.evaluate(init_op)
288 init_op = variables.global_variables_initializer()
289 return graph, init_op, train_op
294 graph, init_op, train_op = self._annotated_graph()
302 self.evaluate(init_op)
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dgmm.py49 def __init__(self, init_op, is_initialized_op, is_chief): argument
50 self._init_op = init_op
161 init_op,
170 init_op, is_initialized, config.is_chief)]
Dkmeans.py79 def __init__(self, init_op, is_initialized_var, is_chief): argument
89 self._init_op = init_op
196 (all_distances, model_predictions, losses, is_initialized, init_op,
216 _InitializeClustersHook(init_op, is_initialized, config.is_chief)
/external/tensorflow/tensorflow/contrib/session_bundle/
Dexporter.py156 init_op=None, argument
211 if init_op:
212 if not isinstance(init_op, ops.Operation):
213 raise TypeError("init_op needs to be an Operation: %s" % init_op)
214 ops.add_to_collection(constants.INIT_OP_KEY, init_op)
/external/tensorflow/tensorflow/contrib/slim/python/slim/
Devaluation_test.py93 init_op = control_flow_ops.group(variables.global_variables_initializer(),
104 init_op.run()
204 init_op = control_flow_ops.group(variables.global_variables_initializer(),
215 init_op.run()
248 init_op = control_flow_ops.group(variables.global_variables_initializer(),
252 sess.run(init_op)
/external/tensorflow/tensorflow/contrib/session_bundle/example/
Dexport_half_plus_two.py118 init_op = tf.group(assign_asset_path, assign_filename1, assign_filename2)
132 init_op=init_op,
/external/tensorflow/tensorflow/contrib/eager/python/
Devaluator.py170 init_op = self.init_variables()
172 return (init_op, call_op, results_op)
175 def run_evaluation(init_op, call_op, results_op, sess=None): argument
201 sess.run(init_op)
Devaluator_test.py123 init_op, call_op, results_op = e.evaluate_on_dataset(ds)
124 results = e.run_evaluation(init_op, call_op, results_op)
134 init_op, call_op, results_op = e.evaluate_on_dataset(
137 e.run_evaluation(init_op, call_op, results_op)
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dkmeans.py87 def __init__(self, init_op, is_initialized_op, is_chief): argument
88 self._init_op = init_op
120 is_initialized, init_op, training_op) = clustering_ops.KMeans(
141 init_op, is_initialized, config.is_chief)]

1234