| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | shard_dataset_op.cc | 48 num_shards_(num_shards), in Dataset()
77 params.set_args(num_shards_, index_); in DebugString()
86 return n / num_shards_ + (index_ < n % num_shards_ ? 1 : 0); in CardinalityInternal()
94 return n / num_shards_ + (index_ < n % num_shards_ ? 1 : 0); in CardinalityInternal()
109 return input_->Get(ctx, index_ + (num_shards_ * index), out_tensors); in Get()
119 TF_RETURN_IF_ERROR(b->AddScalar(num_shards_, &num_shards)); in AsGraphDefInternal()
139 if (dataset()->num_shards_ == kShardHint) { in Initialize()
163 (dataset()->index_ - next_index_) % dataset()->num_shards_; in GetNextInternal()
165 num_to_skip += dataset()->num_shards_; in GetNextInternal()
185 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_t>(TensorShape({}), {{num_shards_}, {index_}}); in GetInputTensors()
64 int64_t num_shards_; member in tensorflow::data::__anond13f28970111::ShardDatasetParams
|
| /external/grpc-grpc/src/core/lib/event_engine/posix_engine/ |
| D | timer.cc | 53 num_shards_(grpc_core::Clamp(2 * gpr_cpu_num_cores(), 1u, 32u)), in TimerList()
55 shards_(new Shard[num_shards_]), in TimerList()
56 shard_queue_(new Shard*[num_shards_]) { in TimerList()
57 for (size_t i = 0; i < num_shards_; i++) { in TimerList()
101 while (shard->shard_queue_index < num_shards_ - 1 && in NoteDeadlineChange()
111 Shard* shard = &shards_[grpc_core::HashPointer(timer, num_shards_)]; in TimerInit()
164 Shard* shard = &shards_[grpc_core::HashPointer(timer, num_shards_)]; in TimerCancel()
|
| D | timer.h | 149 const size_t num_shards_; variable
|
| /external/rust/android-crates-io/crates/grpcio-sys/grpc/src/core/lib/event_engine/posix_engine/ |
| D | timer.cc | 53 num_shards_(grpc_core::Clamp(2 * gpr_cpu_num_cores(), 1u, 32u)), in TimerList()
55 shards_(new Shard[num_shards_]), in TimerList()
56 shard_queue_(new Shard*[num_shards_]) { in TimerList()
57 for (size_t i = 0; i < num_shards_; i++) { in TimerList()
101 while (shard->shard_queue_index < num_shards_ - 1 && in NoteDeadlineChange()
111 Shard* shard = &shards_[grpc_core::HashPointer(timer, num_shards_)]; in TimerInit()
164 Shard* shard = &shards_[grpc_core::HashPointer(timer, num_shards_)]; in TimerCancel()
|
| D | timer.h | 176 const size_t num_shards_; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | range_sampler.cc | 243 num_shards_(num_shards), in FixedUnigramSampler()
259 num_shards_(num_shards), in FixedUnigramSampler()
281 if (word_id % num_shards_ == shard_) weights_.push_back(0.0); in FillReservedIds()
299 if (word_id % num_shards_ == shard_) { in LoadFromFile()
319 if (word_id % num_shards_ == shard_) { in LoadFromUnigrams()
|
| D | range_sampler.h | 238 int32 num_shards_; variable
|
| /external/pytorch/torch/csrc/distributed/autograd/context/ |
| D | container.h | 156 uint32_t num_shards_; variable
|
| D | container.cpp | 28 num_shards_(num_shards), in DistAutogradContainer()
88 return autograd_contexts_[context_id & (num_shards_ - 1)]; in getShard()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | split_utils.cc | 69 : num_shards_(num_shards), in ShardingSplitProvider()
83 num_to_skip_ = num_shards_ - 1; in GetNext()
|
| D | split_utils.h | 60 const int64_t num_shards_;
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.h | 329 int num_shards_; variable
|
| D | tensor_bundle.cc | 802 num_shards_ = header.num_shards(); in BundleReader()
890 DataFilename(prefix_, entry.shard_id(), num_shards_), &file)); in GetValue()
|