| /external/tensorflow/tensorflow/python/keras/estimator/ |
| D | __init__.py | 39 model_dir=None, argument
176 model_dir=model_dir,
188 model_dir=None, argument
364 model_dir=model_dir,
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | checkpoint_input_pipeline_hook_test.py | 64 def _read_vars(self, model_dir): argument
67 ckpt_path = checkpoint_management.latest_checkpoint(model_dir)
87 self.assertSequenceEqual(self._read_vars(est.model_dir), (2, 1))
89 self.assertSequenceEqual(self._read_vars(est.model_dir), (4, 3))
102 self.assertSequenceEqual(self._read_vars(est.model_dir), (2, 1))
104 self.assertSequenceEqual(self._read_vars(est.model_dir), (4, 3))
115 self.assertSequenceEqual(self._read_vars(est.model_dir), (2, 1))
117 self.assertSequenceEqual(self._read_vars(est.model_dir), (4, 3))
120 self.assertSequenceEqual(self._read_vars(est.model_dir), (6, 1))
|
| /external/tensorflow/tensorflow/python/training/ |
| D | basic_session_run_hooks_test.py | 389 self.model_dir = tempfile.mkdtemp()
397 shutil.rmtree(self.model_dir, ignore_errors=True)
402 self.model_dir, save_steps=1)
410 checkpoint_utils.load_variable(self.model_dir,
416 self.model_dir, saver=self.scaffold.saver, scaffold=self.scaffold)
422 self.model_dir, save_secs=10, save_steps=20)
427 basic_session_run_hooks.CheckpointSaverHook(self.model_dir)
432 self.model_dir, save_secs=2, scaffold=self.scaffold)
440 checkpoint_utils.load_variable(self.model_dir,
447 self.model_dir,
[all …]
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | async_checkpoint_test.py | 106 model_dir=os.path.join(FLAGS.model_dir, 'runconfig'),
130 FLAGS.model_dir,
136 FLAGS.model_dir) # pylint: disable=protected-access
140 FLAGS.model_dir + '/model.ckpt*.meta')
156 model_dir=os.path.join(FLAGS.model_dir, 'runconfig'),
178 FLAGS.model_dir,
183 FLAGS.model_dir) # pylint: disable=protected-access
187 FLAGS.model_dir + '/model.ckpt*.meta')
|
| D | tpu_test_wrapper_test.py | 104 self.assertStartsWith(flags.FLAGS.model_dir,
107 len(flags.FLAGS.model_dir), len('gs://example-bucket/tempfiles'))
113 first = flags.FLAGS.model_dir
115 second = flags.FLAGS.model_dir
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | quantization_mnist_test.py | 114 def _GetGraphDef(self, use_trt, max_batch_size, model_dir): argument
133 checkpoint_file = latest_checkpoint(model_dir)
137 'model_dir={} is the right directory'.format(model_dir))
165 def _Run(self, is_training, use_trt, batch_size, num_epochs, model_dir): argument
214 graph_def = self._GetGraphDef(use_trt, batch_size, model_dir)
242 model_dir=model_dir if is_training else None,
264 model_dir = test.test_src_dir_path(
272 model_dir=model_dir)['accuracy']
284 model_dir=model_dir)['accuracy']
|
| /external/tensorflow/tensorflow/python/distribute/integration_test/ |
| D | saved_model_test.py | 506 model_dir, argument
514 session, [tf1.saved_model.tag_constants.SERVING], model_dir)
551 model_dir = self.get_temp_dir()
555 tf.saved_model.save(m, model_dir)
558 loaded = tf.saved_model.load(model_dir)
566 self.assertAllEqual(self.load_and_run_v1(model_dir, {"x": 1}), [6, 6, 6, 6])
571 model_dir = self.get_temp_dir()
575 tf.saved_model.save(m, model_dir)
578 loaded = tf.saved_model.load(model_dir)
586 model_dir = self.get_temp_dir()
[all …]
|
| /external/tensorflow/tensorflow/cc/saved_model/experimental/tests/ |
| D | saved_model_api_test.cc | 61 std::string model_dir = SavedModelPath("VarsAndArithmeticObjectGraph"); in TEST_P() local
64 SavedModelAPI::Load(model_dir, *runtime, &status, &tags); in TEST_P()
85 std::string model_dir = SavedModelPath("VarsAndArithmeticObjectGraph"); in TEST_P() local
87 SavedModelAPI::Load(model_dir, *runtime, &status); in TEST_P()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/internal/ |
| D | saved_model_api_test.cc | 75 std::string model_dir = SavedModelPath("VarsAndArithmeticObjectGraph"); in TEST_P() local
78 TF_LoadSavedModelWithTags(model_dir.c_str(), ctx, kServeTag, 1, status); in TEST_P()
107 std::string model_dir = SavedModelPath("VarsAndArithmeticObjectGraph"); in TEST_P() local
110 TF_LoadSavedModel(model_dir.c_str(), ctx, status); in TEST_P()
205 std::string model_dir = SavedModelPath("VarsAndArithmeticObjectGraph"); in TEST_P() local
208 TF_LoadSavedModel(model_dir.c_str(), ctx, status); in TEST_P()
309 std::string model_dir = SavedModelPath("AssetModule"); in TEST_P() local
312 TF_LoadSavedModel(model_dir.c_str(), ctx, status); in TEST_P()
364 std::string model_dir = SavedModelPath("StaticHashTableModule"); in TEST_P() local
367 TF_LoadSavedModel(model_dir.c_str(), ctx, status); in TEST_P()
[all …]
|
| /external/tensorflow/tensorflow/python/debug/examples/v1/ |
| D | debug_tflearn_iris.py | 51 model_dir = FLAGS.model_dir or tempfile.mkdtemp(prefix="debug_tflearn_iris_")
57 model_dir=model_dir)
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | import_pb_to_tensorboard.py | 43 def import_to_tensorboard(model_dir, log_dir, tag_set): argument
57 input_graph_def = saved_model_utils.get_meta_graph_def(model_dir,
68 import_to_tensorboard(FLAGS.model_dir, FLAGS.log_dir, FLAGS.tag_set)
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/testdata/ |
| D | BUILD | 31 # model_dir = os.environ['TF_LSO_MODEL_DIR']
33 # model.save(model_dir)
|
| /external/tensorflow/tensorflow/lite/testing/kernel_test/ |
| D | input_generator.cc | 86 TfLiteStatus InputGenerator::LoadModel(const string& model_dir) { in LoadModel() argument
87 model_ = FlatBufferModel::BuildFromFile(model_dir.c_str()); in LoadModel()
89 fprintf(stderr, "Cannot load model %s", model_dir.c_str()); in LoadModel()
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/ |
| D | run_models.py | 138 model_dir = os.path.join(self._output_dir, model_name)
139 gfile.MkDir(model_dir)
140 test_dir = os.path.join(model_dir, test_name)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.estimator.-baseline-regressor.pbtxt | 12 name: "model_dir"
25 …argspec: "args=[\'self\', \'model_dir\', \'label_dimension\', \'weight_column\', \'optimizer\', \'…
|
| D | tensorflow.estimator.-linear-regressor.pbtxt | 12 name: "model_dir"
25 …argspec: "args=[\'self\', \'feature_columns\', \'model_dir\', \'label_dimension\', \'weight_column…
|
| D | tensorflow.estimator.-linear-estimator.pbtxt | 12 name: "model_dir"
25 …argspec: "args=[\'self\', \'head\', \'feature_columns\', \'model_dir\', \'optimizer\', \'config\',…
|
| D | tensorflow.estimator.-run-config.pbtxt | 54 name: "model_dir"
111 …argspec: "args=[\'self\', \'model_dir\', \'tf_random_seed\', \'save_summary_steps\', \'save_checkp…
|
| D | tensorflow.estimator.-estimator.pbtxt | 11 name: "model_dir"
24 …argspec: "args=[\'self\', \'model_fn\', \'model_dir\', \'config\', \'params\', \'warm_start_from\'…
|
| D | tensorflow.estimator.-baseline-classifier.pbtxt | 12 name: "model_dir"
25 …argspec: "args=[\'self\', \'model_dir\', \'n_classes\', \'weight_column\', \'label_vocabulary\', \…
|
| D | tensorflow.estimator.-baseline-estimator.pbtxt | 12 name: "model_dir"
25 …argspec: "args=[\'self\', \'head\', \'model_dir\', \'optimizer\', \'config\'], varargs=None, keywo…
|
| D | tensorflow.estimator.-linear-classifier.pbtxt | 12 name: "model_dir"
25 …argspec: "args=[\'self\', \'feature_columns\', \'model_dir\', \'n_classes\', \'weight_column\', \'…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.estimator.-estimator.pbtxt | 10 name: "model_dir"
23 …argspec: "args=[\'self\', \'model_fn\', \'model_dir\', \'config\', \'params\', \'warm_start_from\'…
|
| D | tensorflow.estimator.-run-config.pbtxt | 54 name: "model_dir"
111 …argspec: "args=[\'self\', \'model_dir\', \'tf_random_seed\', \'save_summary_steps\', \'save_checkp…
|
| D | tensorflow.estimator.-linear-regressor.pbtxt | 11 name: "model_dir"
24 …argspec: "args=[\'self\', \'feature_columns\', \'model_dir\', \'label_dimension\', \'weight_column…
|