Searched refs:merged_prefix (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/checkpoint/ |
| D | functional_saver.py | 163 def restore_fn(merged_prefix): argument
164 return fn(trackables=trackables, merged_prefix=merged_prefix)
170 merged_prefix=tensor_spec.TensorSpec(shape=(), dtype=dtypes.string))
172 def restore_fn_with_replaced_captures(merged_prefix): argument
173 return call_with_mapped_captures(concrete, [merged_prefix])
|
| D | save_util_v1_test.py | 34 restore_fn=lambda trackables, merged_prefix: None)
|
| /external/tensorflow/tensorflow/python/saved_model/registration/ |
| D | registration_saving_test.py | 97 def restore_stacks_and_parts(trackables, merged_prefix): argument
101 restored_tensors = io_ops.restore_v2(merged_prefix, tensor_names,
328 def restore_fn(trackables, merged_prefix): argument
329 del merged_prefix # Unused.
379 def restore_fn(trackables, merged_prefix): argument
384 restored_tensors = io_ops.restore_v2(merged_prefix, tensor_names,
391 restored_tensors = io_ops.restore_v2(merged_prefix, tensor_names,
|
| D | README.md | 154 def restore_stacks_and_parts(trackables, merged_prefix):
158 restored_tensors = io_ops.restore_v2(merged_prefix, tensor_names,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_restore_v2_ops.cc | 283 const string& merged_prefix = destination_prefix.scalar<tstring>()(); in Compute() local
285 tensorflow::MergeBundles(env, input_prefixes, merged_prefix, in Compute()
289 const string merged_dir(io::Dirname(merged_prefix)); in Compute()
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.cc | 696 StringPiece merged_prefix, bool allow_missing_files) { in MergeBundles() argument
700 Status status = env->CreateDir(string(io::Dirname(merged_prefix))); in MergeBundles()
720 << DataFilename(merged_prefix, p.second, merge.shard_ids.size()); in MergeBundles()
723 DataFilename(merged_prefix, p.second, merge.shard_ids.size()))); in MergeBundles()
729 env->NewWritableFile(MetaFilename(merged_prefix), &merged_metadata)); in MergeBundles()
746 VLOG(1) << "Merged bundles to:" << merged_prefix; in MergeBundles()
|
| D | tensor_bundle.h | 190 StringPiece merged_prefix,
|