Searched refs:num_shards_ (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | shard_dataset_op.cc | 46 num_shards_(num_shards), in Dataset()
75 params.set_args(num_shards_, index_); in DebugString()
84 return n / num_shards_ + (index_ < n % num_shards_ ? 1 : 0); in Cardinality()
103 TF_RETURN_IF_ERROR(b->AddScalar(num_shards_, &num_shards)); in AsGraphDefInternal()
138 (dataset()->index_ - next_index_) % dataset()->num_shards_; in GetNextInternal()
140 num_to_skip += dataset()->num_shards_; in GetNextInternal()
160 next_index_ < dataset()->num_shards_) { in GetNextInternal()
162 Status s = input_impl_->Skip(ctx, dataset()->num_shards_ - next_index_, in GetNextInternal()
171 dataset()->num_shards_, in GetNextInternal()
183 next_index_ = dataset()->num_shards_; in GetNextInternal()
[all …]
|
| D | shard_dataset_op_test.cc | 31 num_shards_(num_shards), in ShardDatasetParams()
41 return CreateTensors<int64>(TensorShape({}), {{num_shards_}, {index_}}); in GetInputTensors()
64 int64 num_shards_; member in tensorflow::data::__anonae7fb6210111::ShardDatasetParams
|
| D | split_utils.cc | 63 : num_shards_(num_shards), in ShardingSplitProvider()
77 num_to_skip_ = num_shards_ - 1; in GetNext()
|
| D | split_utils.h | 57 const int64 num_shards_;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | range_sampler.cc | 242 num_shards_(num_shards), in FixedUnigramSampler()
258 num_shards_(num_shards), in FixedUnigramSampler()
280 if (word_id % num_shards_ == shard_) weights_.push_back(0.0); in FillReservedIds()
298 if (word_id % num_shards_ == shard_) { in LoadFromFile()
318 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 | 303 int num_shards_; variable
|
| D | tensor_bundle.cc | 789 num_shards_ = header.num_shards(); in BundleReader()
877 DataFilename(prefix_, entry.shard_id(), num_shards_), &file)); in GetValue()
|