Home
last modified time | relevance | path

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

123456

/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Destimator_test.py140 def _assert_checkpoint(self, model_dir, global_step): argument
141 reader = checkpoint_utils.load_checkpoint(model_dir)
148 model_dir = tempfile.mkdtemp()
155 model_dir=model_dir,
167 model_dir = tempfile.mkdtemp()
174 model_dir=model_dir,
189 model_dir = tempfile.mkdtemp()
201 model_dir=model_dir,
214 model_dir = tempfile.mkdtemp()
221 model_dir=model_dir,
[all …]
Destimator.py50 model_dir=None, argument
143 model_dir=model_dir,
159 model_dir=None, argument
236 model_dir=model_dir,
254 model_dir=None, argument
318 model_dir=model_dir,
334 model_dir=None, argument
411 model_dir=model_dir,
430 model_dir=None, argument
513 model_dir=model_dir,
[all …]
Ddnn_tree_combined_estimator_test.py92 model_dir = tempfile.mkdtemp()
102 model_dir=model_dir,
115 model_dir = tempfile.mkdtemp()
125 model_dir=model_dir,
138 def _assert_checkpoint(self, model_dir, global_step): argument
139 reader = checkpoint_utils.load_checkpoint(model_dir)
149 model_dir = tempfile.mkdtemp()
159 model_dir=model_dir,
168 self._assert_checkpoint(est.model_dir, global_step=14)
180 model_dir = tempfile.mkdtemp()
[all …]
Dtrainer_hooks_test.py40 trainer_hooks.FeatureImportanceSummarySaver(model_dir=None)
44 model_dir = tempfile.mkdtemp()
45 hook = trainer_hooks.FeatureImportanceSummarySaver(model_dir)
54 model_dir = tempfile.mkdtemp()
55 hook = trainer_hooks.FeatureImportanceSummarySaver(model_dir)
71 self.assertTrue(os.path.exists(os.path.join(model_dir, "featA")))
72 self.assertTrue(os.path.exists(os.path.join(model_dir, "featB")))
Ddnn_tree_combined_estimator.py401 model_dir=None, argument
512 model_dir=model_dir,
527 model_dir=None, argument
643 model_dir=model_dir,
661 model_dir=None, argument
759 model_dir=model_dir,
819 model_dir=None, argument
859 model_fn=_model_fn, model_dir=model_dir, config=config)
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Drun_config_test.py41 tf_random_seed=RANDOM_SEED, model_dir=TEST_DIR)
229 self.assertIsNone(empty_config.model_dir)
231 config = run_config_lib.RunConfig(model_dir=TEST_DIR)
232 self.assertEqual(TEST_DIR, config.model_dir)
238 self.assertEqual(TEST_DIR, run_config.model_dir)
243 run_config = run_config_lib.RunConfig(model_dir=TEST_DIR)
244 self.assertEqual(TEST_DIR, run_config.model_dir)
253 run_config_lib.RunConfig(model_dir=TEST_DIR + "/sub_dir")
257 tf_random_seed=RANDOM_SEED, model_dir=TEST_DIR)
258 self.assertEqual(TEST_DIR, config.model_dir)
[all …]
Dcomposable_model.py196 def get_weights(self, model_dir): argument
205 all_variables = [name for name, _ in list_variables(model_dir)]
212 values[name] = load_variable(model_dir, name)
217 def get_bias(self, model_dir): argument
226 return load_variable(model_dir, name=(self._scope + "/bias_weight"))
318 def get_weights(self, model_dir): argument
329 model_dir, name=(self._scope + "/hiddenlayer_%d/weights" % i))
332 model_dir, name=(self._scope + "/logits/weights"))]
334 def get_bias(self, model_dir): argument
345 model_dir, name=(self._scope + "/hiddenlayer_%d/biases" % i))
[all …]
Ddebug.py164 model_dir=None, argument
204 model_dir=model_dir,
284 model_dir=None, argument
319 model_dir=model_dir,
Drun_config.py245 model_dir=None, argument
333 self._model_dir = _get_model_dir(model_dir)
371 def model_dir(self): member in RunConfig
468 def _get_model_dir(model_dir): argument
474 if model_dir is not None and model_dir_in_tf_config != model_dir:
479 model_dir, model_dir_in_tf_config))
483 return model_dir or model_dir_in_tf_config
Ddnn.py304 model_dir=None, argument
370 model_dir=model_dir,
593 model_dir=None, argument
653 model_dir=model_dir,
874 model_dir=None, argument
922 model_dir=model_dir,
/external/tensorflow/tensorflow/contrib/learn/python/learn/
Dmonitors_test.py324 model_dir = 'model/dir'
325 estimator.model_dir = model_dir
336 mock_latest_checkpoint.assert_called_with(model_dir)
344 model_dir = 'model/dir'
345 estimator.model_dir = model_dir
347 mock_latest_checkpoint.return_value = '%s/ckpt' % model_dir
363 model_dir = 'model/dir'
364 estimator.model_dir = model_dir
366 mock_latest_checkpoint.return_value = '%s/ckpt' % model_dir
382 model_dir = 'model/dir'
[all …]
Dlearn_runner_test.py63 def __init__(self, default=None, config=None, model_dir=None): argument
66 internal_model_dir = model_dir or config.model_dir
73 def model_dir(self): member in TestExperiment.__init__.Estimator
106 return TestExperiment(model_dir=output_dir)
112 return TestExperiment(config=run_config, model_dir=output_dir)
210 run_config = run_config_lib.RunConfig(model_dir=_MODIR_DIR)
218 run_config = run_config_lib.RunConfig(model_dir=_MODIR_DIR)
227 run_config = run_config_lib.RunConfig(model_dir=_MODIR_DIR)
235 run_config = run_config_lib.RunConfig(model_dir="")
256 run_config = run_config_lib.RunConfig(model_dir=_MODIR_DIR)
[all …]
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
Destimators_test.py52 model_dir = tempfile.mkdtemp(dir=self.get_temp_dir())
56 first_estimator = estimator_fn(model_dir, exogenous_feature_columns)
81 second_estimator = estimator_fn(model_dir, exogenous_feature_columns)
186 def _estimator_fn(model_dir, exogenous_feature_columns): argument
189 num_features=1, model_dir=model_dir, config=_SeedRunConfig(),
197 def _estimator_fn(model_dir, exogenous_feature_columns): argument
207 model_dir=model_dir)
212 def _estimator_fn(model_dir, exogenous_feature_columns): argument
214 num_features=1, periodicities=10, model_dir=model_dir, dtype=dtype,
223 num_features=1, periodicities=[], model_dir=self.get_temp_dir(),
[all …]
Destimators.py50 def __init__(self, model, state_manager=None, optimizer=None, model_dir=None, argument
81 model_dir=model_dir,
304 optimizer=None, model_dir=None, config=None): argument
386 model_dir=model_dir,
478 model_dir=None, argument
541 model_dir=model_dir,
549 def __init__(self, model, state_manager=None, optimizer=None, model_dir=None, argument
562 model_dir=model_dir,
606 model_dir=None, argument
689 model_dir=model_dir,
Dhead_test.py144 model_dir = self.get_temp_dir()
149 model_dir=model_dir)
328 model_dir, head_type, exogenous_feature_columns): argument
337 model_dir=model_dir)
341 model_dir, head_type, exogenous_feature_columns): argument
347 model_dir=model_dir)
351 model_dir, head_type, exogenous_feature_columns): argument
361 model_dir=model_dir)
376 model_dir = _new_temp_dir()
385 model_dir=model_dir,
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dcheckpoint_input_pipeline_hook_test.py58 def _read_vars(self, model_dir): argument
61 ckpt_path = checkpoint_management.latest_checkpoint(model_dir)
80 self.assertSequenceEqual(self._read_vars(est.model_dir), (2, 1))
82 self.assertSequenceEqual(self._read_vars(est.model_dir), (4, 3))
94 self.assertSequenceEqual(self._read_vars(est.model_dir), (2, 1))
96 self.assertSequenceEqual(self._read_vars(est.model_dir), (4, 3))
106 self.assertSequenceEqual(self._read_vars(est.model_dir), (2, 1))
108 self.assertSequenceEqual(self._read_vars(est.model_dir), (4, 3))
111 self.assertSequenceEqual(self._read_vars(est.model_dir), (6, 1))
/external/tensorflow/tensorflow/python/training/
Dbasic_session_run_hooks_test.py390 self.model_dir = tempfile.mkdtemp()
398 shutil.rmtree(self.model_dir, ignore_errors=True)
403 self.model_dir, save_steps=1)
411 checkpoint_utils.load_variable(self.model_dir,
417 self.model_dir, saver=self.scaffold.saver, scaffold=self.scaffold)
423 self.model_dir, save_secs=10, save_steps=20)
428 basic_session_run_hooks.CheckpointSaverHook(self.model_dir)
433 self.model_dir, save_secs=2, scaffold=self.scaffold)
441 checkpoint_utils.load_variable(self.model_dir,
448 self.model_dir,
[all …]
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dquantization_mnist_test.py113 def _GetGraphDef(self, use_trt, max_batch_size, model_dir): argument
132 checkpoint_file = latest_checkpoint(model_dir)
161 def _Run(self, is_training, use_trt, batch_size, num_epochs, model_dir): argument
219 graph_def = self._GetGraphDef(use_trt, batch_size, model_dir)
247 model_dir=model_dir if is_training else None,
268 model_dir = test.test_src_dir_path('python/compiler/tensorrt/test/testdata')
275 model_dir=model_dir)['accuracy']
287 model_dir=model_dir)['accuracy']
/external/tensorflow/tensorflow/contrib/linear_optimizer/python/
Dsdca_estimator.py213 model_dir=None, argument
281 model_dir=model_dir,
341 model_dir=None, argument
382 model_dir=model_dir,
472 model_dir=None, argument
514 model_dir=model_dir,
/external/tensorflow/tensorflow/contrib/kernel_methods/python/
Dkernel_estimators.py157 model_dir=None, argument
183 model_dir=model_dir,
259 model_dir=None, argument
302 model_dir=model_dir,
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dgmm.py77 model_dir=None, argument
104 model_fn=self._model_builder(), model_dir=model_dir, config=config)
132 self.model_dir, gmm_ops.GmmAlgorithm.CLUSTERS_WEIGHT)
137 self.model_dir, gmm_ops.GmmAlgorithm.CLUSTERS_VARIABLE)
143 self.model_dir, gmm_ops.GmmAlgorithm.CLUSTERS_COVS_VARIABLE)
/external/tensorflow/tensorflow/contrib/distribute/python/examples/
Dkeras_model_with_estimator.py42 model_dir = args[1]
43 print('Using %s to store checkpoints.' % model_dir)
63 keras_model=model, config=config, model_dir=model_dir)
/external/tensorflow/tensorflow/python/keras/estimator/
D__init__.py34 model_dir=None, argument
72 model_dir=model_dir,
/external/tensorflow/tensorflow/examples/tf2_showcase/
Dmnist.py201 train_dir = os.path.join(flags_obj.model_dir, 'summaries', 'train')
202 test_dir = os.path.join(flags_obj.model_dir, 'summaries', 'eval')
209 checkpoint_dir = os.path.join(flags_obj.model_dir, 'checkpoints')
238 export_path = os.path.join(flags_obj.model_dir, 'export')
248 if flags_obj.clean and tf.gfile.Exists(flags_obj.model_dir):
250 flags_obj.model_dir))
251 tf.gfile.DeleteRecursively(flags_obj.model_dir)
/external/tensorflow/tensorflow/contrib/framework/python/ops/
Dvariables_test.py964 model_dir = tempfile.mkdtemp(
973 model_dir)
994 model_dir = tempfile.mkdtemp(
1003 model_dir)
1034 model_dir = tempfile.mkdtemp(
1044 model_dir)
1054 model_dir = tempfile.mkdtemp(
1067 model_dir)
1117 model_dir = tempfile.mkdtemp(
1119 if gfile.Exists(model_dir):
[all …]

123456