| /external/tensorflow/tensorflow/python/keras/estimator/ |
| __init__.py | 34 model_dir=None,
49 model_dir: Directory to save `Estimator` model parameters, graph, summary
72 model_dir=model_dir,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| trainer_hooks_test.py | 40 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")))
|
| dnn_tree_combined_estimator_test.py | 92 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):
139 reader = checkpoint_utils.load_checkpoint(model_dir)
149 model_dir = tempfile.mkdtemp()
159 model_dir=model_dir
[all...] |
| estimator_test.py | 140 def _assert_checkpoint(self, model_dir, global_step):
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
[all...] |
| estimator.py | 50 model_dir=None,
72 model_dir: Directory for model exports, etc.
143 model_dir=model_dir,
159 model_dir=None,
184 model_dir: Directory for model exports, etc.
236 model_dir=model_dir,
254 model_dir=None,
275 model_dir: Directory for model exports, etc
[all...] |
| /external/tensorflow/tensorflow/contrib/distribute/python/examples/ |
| keras_model_with_estimator.py | 38 print('You must specify model_dir for checkpoints such as'
42 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/lite/testing/kernel_test/ |
| input_generator.h | 35 TfLiteStatus LoadModel(const string& model_dir);
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| run_config_test.py | 41 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)
235 tf_config = {"model_dir": TEST_DIR}
238 self.assertEqual(TEST_DIR, run_config.model_dir)
241 tf_config = {"model_dir": TEST_DIR}
243 run_config = run_config_lib.RunConfig(model_dir=TEST_DIR)
244 self.assertEqual(TEST_DIR, run_config.model_dir)
247 tf_config = {"model_dir": TEST_DIR
[all...] |
| debug.py | 164 model_dir=None,
173 model_dir: Directory to save model parameters, graph and etc. This can
204 model_dir=model_dir,
284 model_dir=None,
292 model_dir: Directory to save model parameters, graph and etc. This can
319 model_dir=model_dir,
|
| composable_model.py | 196 def get_weights(self, model_dir):
200 model_dir: Directory where model parameters, graph and etc. are saved.
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):
221 model_dir: Directory where model parameters, graph and etc. are saved.
226 return load_variable(model_dir, name=(self._scope + "/bias_weight"))
318 def get_weights(self, model_dir):
322 model_dir: Directory where model parameters, graph and etc. are saved.
329 model_dir, name=(self._scope + "/hiddenlayer_%d/weights" % i)
[all...] |
| run_config.py | 245 model_dir=None,
286 model_dir: directory where model parameters, graph etc are saved. If
287 `None`, will use `model_dir` property in `TF_CONFIG` environment
333 self._model_dir = _get_model_dir(model_dir)
371 def model_dir(self): member in class:RunConfig
468 def _get_model_dir(model_dir):
469 """Returns `model_dir` based user provided `model_dir` or `TF_CONFIG`."""
472 os.environ.get('TF_CONFIG') or '{}').get('model_dir', None)
474 if model_dir is not None and model_dir_in_tf_config != model_dir
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| learn_runner_test.py | 41 "Must specify a model directory `model_dir` in `run_config`.")
63 def __init__(self, default=None, config=None, model_dir=None):
66 internal_model_dir = model_dir or config.model_dir
73 def model_dir(self): member in class: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
[all...] |
| monitors_test.py | 324 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_di
[all...] |
| evaluable.py | 41 def model_dir(self): member in class:Evaluable
112 latest checkpoint in `model_dir` is used.
|
| /external/tensorflow/tensorflow/contrib/predictor/ |
| contrib_estimator_predictor_test.py | 39 model_dir = tempfile.mkdtemp()
41 core=False, model_dir=model_dir)
|
| core_estimator_predictor_test.py | 39 model_dir = tempfile.mkdtemp()
41 core=True, model_dir=model_dir)
|
| /external/tensorflow/tensorflow/examples/saved_model/integration_tests/ |
| saved_model_part1_test.py | 50 self.assertCommandSucceeded(use_binary, model_dir=export_dir)
60 self.assertCommandSucceeded(use_binary, model_dir=export_dir)
71 self.assertCommandSucceeded(use_binary, model_dir=export_dir)
|
| use_rnn_cell.py | 28 flags.DEFINE_string("model_dir", None, "Directory to load SavedModel from.")
33 cell = tf.saved_model.load(FLAGS.model_dir)
|
| use_text_rnn_model.py | 27 flags.DEFINE_string("model_dir", None, "Directory to load SavedModel from.")
37 model = tf.saved_model.load(FLAGS.model_dir)
|
| /external/tensorflow/tensorflow/python/tools/ |
| import_pb_to_tensorboard.py | 43 def import_to_tensorboard(model_dir, log_dir):
47 model_dir: The location of the protobuf (`pb`) model to visualize
56 with gfile.GFile(model_dir, "rb") as f:
68 import_to_tensorboard(FLAGS.model_dir, FLAGS.log_dir)
74 "--model_dir",
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| estimators_test.py | 52 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):
189 num_features=1, model_dir=model_dir, config=_SeedRunConfig(),
197 def _estimator_fn(model_dir, exogenous_feature_columns):
207 model_dir=model_dir)
212 def _estimator_fn(model_dir, exogenous_feature_columns)
[all...] |
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| checkpoint_input_pipeline_hook_test.py | 58 def _read_vars(self, model_dir):
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/contrib/linear_optimizer/python/ |
| sdca_estimator.py | 213 model_dir=None,
232 model_dir: Directory to save model parameters, graph etc. This can also be
281 model_dir=model_dir,
341 model_dir=None,
359 model_dir: Directory to save model parameters, graph etc. This can also be
382 model_dir=model_dir,
472 model_dir=None,
491 model_dir: Directory to save model parameters, graph etc. This can also b
[all...] |
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| quantization_mnist_test.py | 113 def _GetGraphDef(self, use_trt, max_batch_size, model_dir):
119 model_dir: the model directory to load the checkpoints.
132 checkpoint_file = latest_checkpoint(model_dir)
161 def _Run(self, is_training, use_trt, batch_size, num_epochs, model_dir):
173 model_dir: where to save or load checkpoint.
219 graph_def = self._GetGraphDef(use_trt, batch_size, model_dir)
247 model_dir=model_dir if is_training else None,
256 # To generate the checkpoint, set a different model_dir and call self._Run()
258 # model_dir = '/tmp/quantization_mnist
[all...] |
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| gmm.py | 77 model_dir=None,
87 model_dir: the directory to save the model results and log files.
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)
|
