Home
last modified time | relevance | path

Searched refs:GetShardCount (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/mindrecord/io/
Dshard_index_generator.cc527 CHECK_FAIL_RETURN_UNEXPECTED(shard_header_.GetShardCount() <= kMaxShardCount, in WriteToDatabase()
528 "num shards: " + std::to_string(shard_header_.GetShardCount()) + in WriteToDatabase()
536 …:thread::hardware_concurrency() / 2 + 1, static_cast<unsigned int>(shard_header_.GetShardCount())); in WriteToDatabase()
554 while (shard_no < shard_header_.GetShardCount()) { in DatabaseWriter()
Dshard_reader.cc289 int ShardReader::GetShardCount() const { return shard_header_->GetShardCount(); } in GetShardCount() function in mindspore::mindrecord::ShardReader
295 int shard_count = shard_header_->GetShardCount(); in ReadRowGroupSummary()
/third_party/mindspore/mindspore/ccsrc/minddata/mindrecord/include/
Dshard_header.h116 int GetShardCount() const { return shard_count_; } in GetShardCount() function
Dshard_reader.h114 int GetShardCount() const;