Searched refs:shard_dirs (Results 1 – 2 of 2) sorted by relevance
576 const std::vector<std::string>& shard_dirs, in MakeNestedDataset() argument584 datasets.reserve(shard_dirs.size()); in MakeNestedDataset()585 for (const auto& shard_dir : shard_dirs) { in MakeNestedDataset()588 int64 dataset_start_index = start_index / shard_dirs.size(); in MakeNestedDataset()589 if (start_index % shard_dirs.size() > datasets.size()) { in MakeNestedDataset()603 if (!shard_dirs.empty()) { in MakeNestedDataset()605 datasets.begin() + (start_index % shard_dirs.size()), in MakeNestedDataset()
183 const std::vector<std::string>& shard_dirs,