Searched refs:num_shards_ (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | shard_dataset_op.cc | 60 num_shards_(num_shards), in Dataset()
83 return strings::StrCat("ShardDatasetOp(", num_shards_, ", ", index_, in DebugString()
92 return n / num_shards_ + (index_ < n % num_shards_ ? 1 : 0); in Cardinality()
102 TF_RETURN_IF_ERROR(b->AddScalar(num_shards_, &num_shards)); in AsGraphDefInternal()
139 } while ((next_index_++ % dataset()->num_shards_) != dataset()->index_); in GetNextInternal()
149 dataset()->num_shards_); in CreateNode()
184 const int64 num_shards_; member in tensorflow::data::__anon3d7918cb0111::ShardDatasetOp::Dataset
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | range_sampler.cc | 239 num_shards_(num_shards), in FixedUnigramSampler()
255 num_shards_(num_shards), in FixedUnigramSampler()
277 if (word_id % num_shards_ == shard_) weights_.push_back(0.0); in FillReservedIds()
295 if (word_id % num_shards_ == shard_) { in LoadFromFile()
315 if (word_id % num_shards_ == shard_) { in LoadFromUnigrams()
|
| D | range_sampler.h | 238 int32 num_shards_; variable
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.h | 301 int num_shards_; variable
|
| D | tensor_bundle.cc | 749 num_shards_ = header.num_shards(); in BundleReader()
830 DataFilename(prefix_, entry.shard_id(), num_shards_), &file)); in GetValue()
|