Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/training/
Dsession_manager.py239 init_fn=None): argument
297 if init_op is None and not init_fn and self._local_init_op is None:
302 if init_fn:
303 init_fn(sess)
310 init_fn,
318 (_maybe_name(init_op), init_fn, self._local_init_op, msg))
Dmonitored_session.py110 init_fn=None, argument
163 … init_fn = coalesce(init_fn, copy_from_scaffold._user_init_fn) # pylint: disable=protected-access
175 self._user_init_fn = init_fn
176 if init_fn:
177 self._init_fn = lambda sess: init_fn(self, sess)
250 def init_fn(self): member in Scaffold
669 init_fn=self._scaffold.init_fn)
Dmonitored_session_test.py72 self.assertEqual(None, scaffold.init_fn)
87 self.assertEqual(None, scaffold.init_fn)
110 self.assertEqual(None, scaffold.init_fn)
146 init_fn=lambda scaffold, sess: 4,
155 self.assertTrue(callable(scaffold.init_fn))
178 init_fn=lambda scaffold, sess: 4,
189 self.assertTrue(callable(scaffold2.init_fn))
203 init_fn=lambda scaffold, sess: 4,
213 init_fn=lambda scaffold, sess: 8,
224 self.assertTrue(callable(scaffold2.init_fn))
[all …]
Dsupervisor.py228 init_fn=None, argument
329 self._init_fn = init_fn
734 init_fn=self._init_fn)
Dcheckpoint_management.py547 init_fn=None): argument
630 self._init_fn = init_fn
Dsession_manager_test.py75 "", init_fn=lambda sess: sess.run(v.initializer))
711 "", init_fn=lambda sess: sess.run(v.initializer))
Dcheckpoint_management_test.py593 init_fn=lambda: checkpoint.restore(init_path).run_restore_ops())
Dsupervisor_test.py529 sv = supervisor.Supervisor(logdir=logdir, init_op=None, init_fn=_init_fn)
/external/strace/tests/
Dbpf.c101 void (*init_fn)(struct bpf_attr_check *check); member
179 if (check->init_fn) in test_bpf()
180 check->init_fn((struct bpf_attr_check *) check); in test_bpf()
421 .init_fn = init_BPF_MAP_CREATE_attr7,
649 .init_fn = init_BPF_PROG_LOAD_attr3,
665 .init_fn = init_BPF_PROG_LOAD_attr4,
694 .init_fn = init_BPF_OBJ_PIN_attr,
703 .init_fn = init_BPF_OBJ_PIN_attr,
1021 .init_fn = init_BPF_PROG_QUERY_attr4,
1032 .init_fn = init_BPF_PROG_QUERY_attr5,
[all …]
/external/strace/tests-m32/
Dbpf.c101 void (*init_fn)(struct bpf_attr_check *check); member
179 if (check->init_fn) in test_bpf()
180 check->init_fn((struct bpf_attr_check *) check); in test_bpf()
421 .init_fn = init_BPF_MAP_CREATE_attr7,
649 .init_fn = init_BPF_PROG_LOAD_attr3,
665 .init_fn = init_BPF_PROG_LOAD_attr4,
694 .init_fn = init_BPF_OBJ_PIN_attr,
703 .init_fn = init_BPF_OBJ_PIN_attr,
1021 .init_fn = init_BPF_PROG_QUERY_attr4,
1032 .init_fn = init_BPF_PROG_QUERY_attr5,
[all …]
/external/strace/tests-mx32/
Dbpf.c101 void (*init_fn)(struct bpf_attr_check *check); member
179 if (check->init_fn) in test_bpf()
180 check->init_fn((struct bpf_attr_check *) check); in test_bpf()
421 .init_fn = init_BPF_MAP_CREATE_attr7,
649 .init_fn = init_BPF_PROG_LOAD_attr3,
665 .init_fn = init_BPF_PROG_LOAD_attr4,
694 .init_fn = init_BPF_OBJ_PIN_attr,
703 .init_fn = init_BPF_OBJ_PIN_attr,
1021 .init_fn = init_BPF_PROG_QUERY_attr4,
1032 .init_fn = init_BPF_PROG_QUERY_attr5,
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.-scaffold.pbtxt10 name: "init_fn"
43 …argspec: "args=[\'self\', \'init_op\', \'init_feed_dict\', \'init_fn\', \'ready_op\', \'ready_for_…
Dtensorflow.train.-checkpoint-manager.pbtxt27 …urs\', \'checkpoint_name\', \'step_counter\', \'checkpoint_interval\', \'init_fn\'], varargs=None,…
Dtensorflow.train.-session-manager.pbtxt11 …t_for_checkpoint\', \'max_wait_secs\', \'config\', \'init_feed_dict\', \'init_fn\'], varargs=None,…
Dtensorflow.train.-supervisor.pbtxt107 …s\', \'checkpoint_basename\', \'session_manager\', \'summary_writer\', \'init_fn\', \'local_init_r…
/external/tensorflow/tensorflow/python/saved_model/
Dload_v1_in_v2.py57 def __init__(self, init_fn, asset_paths): argument
60 self._init_fn = init_fn
246 init_fn = wrapped.prune(
249 initializer = _Initializer(init_fn, asset_paths)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.train.-checkpoint-manager.pbtxt27 …urs\', \'checkpoint_name\', \'step_counter\', \'checkpoint_interval\', \'init_fn\'], varargs=None,…
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dgroup_by_reducer_test.py182 def init_fn(_): function
193 reducer = grouping.Reducer(init_fn, reduce_fn, finalize_fn)
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.cc755 auto init_fn = reinterpret_cast<SEInitPluginFn>(dso_symbol); in InitStreamExecutorPlugin() local
756 return InitStreamExecutorPlugin(init_fn); in InitStreamExecutorPlugin()
759 port::Status InitStreamExecutorPlugin(SEInitPluginFn init_fn) { in InitStreamExecutorPlugin() argument
771 init_fn(&params, c_status.get()); in InitStreamExecutorPlugin()
Dstream_executor_internal.h39 port::Status InitStreamExecutorPlugin(SEInitPluginFn init_fn);
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dmulti_device_iterator_test.py410 def init_fn(n): function
425 init_fn,
Diterator_test.py936 def init_fn(n): function
950 dataset = dataset_ops._GeneratorDataset(1, init_fn, next_fn, finalize_fn,
/external/tensorflow/tensorflow/python/eager/
Dops_test.py425 def init_fn(): function
436 t1 = threading.Thread(target=init_fn)
Ddef_function_test.py152 init_fn = fn.get_initialization_function(constant_op.constant(1.0))
156 init_fn()
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy_test.py443 def init_fn(): function
446 prev = strategy.run(init_fn)