Searched refs:shard_directory (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | snapshot_util.h | 69 std::string GetCheckpointFileName(const std::string& shard_directory,
323 const std::string& shard_directory, uint64 checkpoint_id,
339 Status WriterThread(Env* env, const std::string& shard_directory,
|
| D | snapshot_util.cc | 74 std::string GetCheckpointFileName(const std::string& shard_directory, in GetCheckpointFileName() argument
77 shard_directory, in GetCheckpointFileName()
939 const std::string& shard_directory, in AsyncWriter() argument
945 [this, env, shard_directory, checkpoint_id, compression, version, in AsyncWriter()
947 done(WriterThread(env, shard_directory, checkpoint_id, compression, in AsyncWriter()
975 Status AsyncWriter::WriterThread(Env* env, const std::string& shard_directory, in WriterThread() argument
980 TF_RETURN_IF_ERROR(env->RecursivelyCreateDir(shard_directory)); in WriterThread()
983 env, GetCheckpointFileName(shard_directory, checkpoint_id), compression, in WriterThread()
|