Searched refs:snapshot_util (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | snapshot_dataset_op.cc | 188 snapshot_util::Mode mode_ TF_GUARDED_BY(mu_);
269 absl::flat_hash_map<int64, std::unique_ptr<snapshot_util::AsyncWriter>>
432 snapshot_util::HashDirectory(dataset()->path_, dataset()->hash_)) {} in Iterator()
495 TF_RETURN_IF_ERROR(snapshot_util::ReadMetadataFile( in InitializeIterator()
506 mode_ = snapshot_util::Mode(iterator_mode); in InitializeIterator()
512 TF_RETURN_IF_ERROR(snapshot_util::ReadMetadataFile( in InitializeIterator()
522 TF_RETURN_IF_ERROR(snapshot_util::DetermineOpState( in InitializeIterator()
528 case snapshot_util::READER: in InitializeIterator()
534 case snapshot_util::WRITER: in InitializeIterator()
538 case snapshot_util::PASSTHROUGH: in InitializeIterator()
[all …]
|
| D | io_ops.cc | 53 auto run_dir = snapshot_util::RunDirectory(path, run_id); in DoCompute()
96 absl::flat_hash_map<int64, std::unique_ptr<snapshot_util::AsyncWriter>> in WriteData()
118 snapshot_util::ShardDirectory(run_dir, shard_index); in WriteData()
119 auto writer_thread = std::make_unique<snapshot_util::AsyncWriter>( in WriteData()
175 return snapshot_util::WriteMetadataFile(env, path, &metadata); in WriteMetadataFile()
292 auto run_dir = snapshot_util::RunDirectory(dataset()->path_, in InitializeInput()
299 snapshot_util::kShardDirectorySuffix)), in InitializeInput()
304 TF_RETURN_IF_ERROR(snapshot_util::Reader::MakeNestedDataset( in InitializeInput()
366 OP_REQUIRES_OK(ctx, snapshot_util::ReadMetadataFile( in MakeDataset()
|
| D | BUILD | 284 ":snapshot_util",
574 name = "snapshot_util",
575 srcs = ["snapshot_util.cc"],
576 hdrs = ["snapshot_util.h"],
597 ":snapshot_util",
610 ":snapshot_util",
|
| D | snapshot_util_test.cc | 28 namespace snapshot_util { namespace
|
| D | snapshot_util.cc | 45 namespace snapshot_util { namespace
979 std::unique_ptr<snapshot_util::Writer> writer; in WriterThread()
982 TF_RETURN_IF_ERROR(snapshot_util::Writer::Create( in WriterThread()
|
| D | snapshot_util.h | 44 namespace snapshot_util {
|