diff options
Diffstat (limited to 'tensorflow/core/kernels/save_restore_v2_ops.cc')
| -rw-r--r-- | tensorflow/core/kernels/save_restore_v2_ops.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/save_restore_v2_ops.cc b/tensorflow/core/kernels/save_restore_v2_ops.cc index 3acf290ea2..ab4de6c815 100644 --- a/tensorflow/core/kernels/save_restore_v2_ops.cc +++ b/tensorflow/core/kernels/save_restore_v2_ops.cc @@ -220,9 +220,9 @@ class MergeV2Checkpoints : public OpKernel { context, tensorflow::MergeBundles(env, input_prefixes, merged_prefix)); if (delete_old_dirs_) { - const string& merged_dir = io::Dirname(merged_prefix).ToString(); + const string& merged_dir = std::string(io::Dirname(merged_prefix)); for (const string& input_prefix : input_prefixes) { - const string& dirname = io::Dirname(input_prefix).ToString(); + const string& dirname = std::string(io::Dirname(input_prefix)); if (dirname == merged_dir) continue; Status status = env->DeleteDir(dirname); // For sharded save, only the first delete will go through and all |