Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/data/kernel_tests/
Dcheckpoint_test.py81 init_op = iterator.initializer
85 return init_op, get_next, save_op, restore_op
92 init_op, get_next, save_op, _ = _build_graph(start, stop)
95 sess.run(init_op)
101 init_op, get_next, _, restore_op = _build_graph(start, stop)
103 sess.run(init_op)
112 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
115 sess.run(init_op)
119 sess.run(init_op)
135 init_op = iterator.initializer
[all …]
Dfrom_sparse_tensor_slices_test.py44 init_op = iterator.initializer
57 sess.run(init_op, feed_dict={st: sparse_feed})
75 sess.run(init_op, feed_dict={st: sparse_feed})
83 sess.run(init_op, feed_dict={st: sparse_feed})
Diterator_cluster_test.py141 init_op = iterator.initializer
146 sess.run(init_op)
174 init_op = iterator.initializer
178 sess.run(init_op)
Dflat_map_test.py79 init_op = iterator.initializer
91 sess.run(init_op)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Ddataset_serialization_test_base.py228 init_op, get_next_op = self._get_iterator_ops_from_collection(
232 self._initialize(init_op, sess)
234 self._initialize(init_op, sess)
263 init_op, get_next_op, saver = self._build_graph(
267 self._initialize(init_op, sess)
352 init_op, get_next_op = self._get_iterator_ops_from_collection(
355 init_op, get_next_op, saver = self._build_graph(
357 return init_op, get_next_op, saver
361 init_op, get_next_op, saver = get_ops()
365 self._initialize(init_op, sess)
[all …]
Drange_dataset_serialization_test.py68 init_op = iterator.initializer
72 return init_op, get_next, save_op, restore_op
79 init_op, get_next, save_op, _ = _build_graph(start, stop)
82 self.evaluate(init_op)
88 init_op, get_next, _, restore_op = _build_graph(start, stop)
90 self.evaluate(init_op)
99 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
102 self.evaluate(init_op)
Dserialization_integration_test.py49 init_op, get_next_op = self._build_input_pipeline(name, num_outputs)
50 init_ops.append(init_op)
/external/tensorflow/tensorflow/python/training/
Dsupervisor.py214 init_op=USE_DEFAULT, argument
316 self._init_init_op(init_op=init_op, init_feed_dict=init_feed_dict)
418 def _init_init_op(self, init_op=USE_DEFAULT, init_feed_dict=None): argument
427 if init_op is Supervisor.USE_DEFAULT:
428 init_op = self._get_first_op_from_collection(ops.GraphKeys.INIT_OP)
429 if init_op is None:
430 init_op = variables.global_variables_initializer()
431 ops.add_to_collection(ops.GraphKeys.INIT_OP, init_op)
432 self._init_op = init_op
533 def init_op(self): member in Supervisor
[all …]
Dsession_manager.py231 init_op=None, argument
297 if init_op is None and not init_fn and self._local_init_op is None:
300 if init_op is not None:
301 sess.run(init_op, feed_dict=init_feed_dict)
309 "Init op: %s, init fn: %s, error: %s" % (_maybe_name(init_op),
318 (_maybe_name(init_op), init_fn, self._local_init_op, msg))
Dsession_manager_test.py54 "", init_op=variables.global_variables_initializer())
65 init_op=variables.global_variables_initializer(),
107 init_op=variables.global_variables_initializer(),
130 init_op=None,
141 init_op=None,
499 sess = sm2.prepare_session("", init_op=v.initializer)
553 sess = sm2.prepare_session("", init_op=None)
594 sess = sm.prepare_session("", init_op=v.initializer)
616 sm2.prepare_session("", init_op=v.initializer)
633 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()
Dquantize_training_test.py67 init_op = variables.global_variables_initializer()
78 self.evaluate(g.get_operation_by_name(init_op.name))
Dbasic_session_run_hooks_test.py406 sess.run(self.scaffold.init_op)
436 sess.run(self.scaffold.init_op)
454 sess.run(self.scaffold.init_op)
587 sess.run(self.scaffold.init_op)
634 sess.run(self.scaffold.init_op)
674 sess.run(self.scaffold.init_op)
688 sess.run(self.scaffold.init_op)
719 sess.run(self.scaffold.init_op)
739 sess.run(self.scaffold.init_op)
762 sess.run(self.scaffold.init_op)
[all …]
Dmonitored_session.py108 init_op=None, argument
159 init_op = coalesce(init_op, copy_from_scaffold.init_op)
181 self._init_op = init_op
254 def init_op(self): member in Scaffold
667 init_op=self._scaffold.init_op,
/external/tensorflow/tensorflow/python/saved_model/
Dloader_impl.py184 init_op = None
190 init_op = ops.get_collection(init_op_key)[0]
191 return init_op
432 init_op = get_init_op(meta_graph_def, import_scope)
433 if init_op is not None:
434 sess.run(fetches=[init_op], feed_dict=asset_tensors_dictionary)
Dbuilder_impl.py235 init_op=None, argument
279 _add_op_to_signature_def_map(signature_def_map, init_op,
308 init_op=None, argument
357 _add_op_to_signature_def_map(signature_def_map, init_op,
/external/tensorflow/tensorflow/python/grappler/
Dmemory_optimizer_test.py132 init_op = variables.global_variables_initializer()
134 return (metagraph, init_op.name, train_op.name, loss.name)
235 init_op = graph.get_operation_by_name(init_op_name)
239 self.evaluate(init_op)
289 init_op = variables.global_variables_initializer()
290 return graph, init_op, train_op
295 graph, init_op, train_op = self._annotated_graph()
303 self.evaluate(init_op)
/external/tensorflow/tensorflow/tools/graph_transforms/
Dsparsify_gather.cc451 NodeDef* init_op = nullptr; in SparsifyGatherInternal() local
455 init_op = replaced_graph_def.mutable_node(i); in SparsifyGatherInternal()
459 if (!init_op) { in SparsifyGatherInternal()
461 init_op = replaced_graph_def.mutable_node()->Add(); in SparsifyGatherInternal()
462 init_op->set_op("NoOp"); in SparsifyGatherInternal()
463 init_op->set_name(group_init_node); in SparsifyGatherInternal()
467 AddNodeInput(StrCat("^", name), init_op); in SparsifyGatherInternal()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/
Dcommon_v1.py89 signature_def_map, init_op, assets_collection = create_signature()
97 main_op=init_op,
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.-scaffold.pbtxt14 name: "init_op"
43 …argspec: "args=[\'self\', \'init_op\', \'init_feed_dict\', \'init_fn\', \'ready_op\', \'ready_for_…
Dtensorflow.train.-session-manager.pbtxt11 …argspec: "args=[\'self\', \'master\', \'init_op\', \'saver\', \'checkpoint_dir\', \'checkpoint_fil…
Dtensorflow.train.-supervisor.pbtxt22 name: "init_op"
107 …', \'graph\', \'ready_op\', \'ready_for_local_init_op\', \'is_chief\', \'init_op\', \'init_feed_di…
/external/tensorflow/tensorflow/compiler/aot/tests/
Dmake_test_graphs.py59 init_op = variables.global_variables_initializer()
62 sess.run(init_op)
74 init_op = variables.global_variables_initializer()
77 sess.run(init_op)
/external/tensorflow/tensorflow/python/ops/
Dlookup_ops.py491 init_op = gen_lookup_ops.initialize_table_from_dataset(
493 ops.add_to_collection(ops.GraphKeys.TABLE_INITIALIZERS, init_op)
494 return init_op
559 init_op = gen_lookup_ops.lookup_table_import_v2(table.resource_handle,
561 ops.add_to_collection(ops.GraphKeys.TABLE_INITIALIZERS, init_op)
562 return init_op
750 init_op = gen_lookup_ops.initialize_table_from_text_file_v2(
754 ops.add_to_collection(ops.GraphKeys.TABLE_INITIALIZERS, init_op)
760 return init_op
/external/libmpeg2/fuzzer/
Dmpeg2_dec_fuzzer.cpp174 impeg2d_init_op_t init_op; in createCodec() local
186 init_op.s_ivd_init_op_t.u4_size = sizeof(impeg2d_init_op_t); in createCodec()
188 ret = ivd_api_function(mCodec, (void *)&init_ip, (void *)&init_op); in createCodec()

12