Home
last modified time | relevance | path

Searched refs:ckpt_path (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Dcheckpoint_ops.py33 def _load_and_remap_matrix(ckpt_path, argument
182 ckpt_path=ckpt_path,
206 def _load_and_remap_matrix_initializer(ckpt_path, argument
401 ckpt_path=ckpt_path,
419 def _load_embedding_initializer(ckpt_path, argument
474 ckpt_path=ckpt_path,
Dcheckpoint_ops_test.py103 ckpt_path=[self.checkpoint_file],
133 ckpt_path=[self.checkpoint_file],
170 ckpt_path=[self.checkpoint_file],
203 ckpt_path=[self.checkpoint_file],
240 ckpt_path=[self.checkpoint_file],
275 ckpt_path=[self.checkpoint_file],
320 ckpt_path=[self.checkpoint_file],
359 ckpt_path=[self.checkpoint_file],
Dcheckpoint_management_test.py239 ckpt_path = os.path.join(save_dir, "checkpoint")
240 ckpt_file = open(ckpt_path, "w")
249 ckpt_path = os.path.join(save_dir, "checkpoint")
250 ckpt_file = open(ckpt_path, "w")
Dwarm_starting_util.py297 ckpt_path=checkpoint_utils._get_checkpoint_filename(prev_ckpt),
/external/tensorflow/tensorflow/python/kernel_tests/
Dcheckpoint_ops_test.py139 ckpt_path=[self.bundle_file],
154 ckpt_path=[self.bundle_file],
169 ckpt_path=[self.bundle_file],
184 ckpt_path=[self.bundle_file],
204 ckpt_path=[self.bundle_file],
222 ckpt_path=[self.bundle_file],
244 ckpt_path=[self.bundle_file],
256 ckpt_path=[self.bundle_file],
270 ckpt_path=[self.bundle_file],
285 ckpt_path=[self.bundle_file],
[all …]
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_tensor_test.cc43 string ckpt_path = io::JoinPath(testing::TensorFlowSrcRoot(), in TFProfTensorTest() local
47 new checkpoint::CheckpointReader(ckpt_path, status)); in TFProfTensorTest()
Dtfprof_show_test.cc65 string ckpt_path = io::JoinPath(testing::TensorFlowSrcRoot(), in TFProfShowTest() local
69 new checkpoint::CheckpointReader(ckpt_path, status)); in TFProfShowTest()
Dtfprof_stats_test.cc59 string ckpt_path = io::JoinPath(testing::TensorFlowSrcRoot(), in TFProfStatsTest() local
63 new checkpoint::CheckpointReader(ckpt_path, status)); in TFProfStatsTest()
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dcheckpoint_input_pipeline_hook_test.py67 ckpt_path = checkpoint_management.latest_checkpoint(model_dir)
68 meta_filename = ckpt_path + '.meta'
72 saver.restore(sess, ckpt_path)
/external/tensorflow/tensorflow/core/kernels/
Dload_and_remap_matrix_op.cc126 const string& ckpt_path = ckpt_path_t->scalar<tstring>()(); in Compute() local
132 LOG(INFO) << "Processing checkpoint : " << ckpt_path; in Compute()
133 BundleReader reader(context->env(), ckpt_path); in Compute()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DLoadAndRemapMatrix.pbtxt4 name: "ckpt_path"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DLoadAndRemapMatrix.pbtxt4 name: "ckpt_path"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_LoadAndRemapMatrix.pbtxt4 name: "ckpt_path"
70 at `ckpt_path` and potentially reorders its rows and columns using the
/external/tensorflow/tensorflow/python/saved_model/
Dload_v1_in_v2_test.py639 ckpt_path = os.path.join(self.get_temp_dir(), "restore_ckpt")
640 saver.Saver().save(session, ckpt_path)
646 self.evaluate(imported.restore(ckpt_path))
/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Dsaved_model_test.py317 ckpt_path = os.path.join(self.get_temp_dir(), 'weights')
318 loaded.save_weights(ckpt_path)
321 model.load_weights(ckpt_path)
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining.py2877 ckpt_path = os.path.join(filepath, sm_constants.VARIABLES_DIRECTORY,
2879 if _is_readable_tf_checkpoint(ckpt_path):
2880 filepath = ckpt_path
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_v2_test.py5077 ckpt_path = test.test_src_dir_path(
5099 ckpt_to_load_from=ckpt_path,
Dfeature_column_test.py5193 ckpt_path = test.test_src_dir_path(
5215 ckpt_to_load_from=ckpt_path,
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt2133 …argspec: "args=[\'ckpt_path\', \'old_tensor_name\', \'row_remapping\', \'col_remapping\', \'initia…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt2133 …argspec: "args=[\'ckpt_path\', \'old_tensor_name\', \'row_remapping\', \'col_remapping\', \'initia…
/external/tensorflow/tensorflow/go/op/
Dwrappers.go26960 func LoadAndRemapMatrix(scope *Scope, ckpt_path tf.Output, old_tensor_name tf.Output, row_remapping…
26971 ckpt_path, old_tensor_name, row_remapping, col_remapping, initializing_values,
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt21324 name: "ckpt_path"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt31856 name: "ckpt_path"