Searched refs:even_dist_ (Results 1 – 6 of 6) sorted by relevance
42 even_dist_(even_dist) { in DistributedSamplerObj()80 offset_, even_dist_); in SamplerBuild()105 args["even_dist"] = even_dist_; in to_json()135 …DistributedSamplerObj>(num_shards_, shard_id_, shuffle_, num_samples_, seed_, offset_, even_dist_); in SamplerCopy()
81 bool even_dist_; variable
33 even_dist_(even_dist), in DistributedSamplerRT()62 if (offset_ != -1 || !even_dist_) { in InitSampler()99 } else if (cnt_ == samples_per_tensor_ && (non_empty_ || !even_dist_)) { in GetNextSample()183 if (offset_ != -1 || !even_dist_) { in CalculateNumSamples()
88 bool even_dist_; variable
40 even_dist_(even_dist) {} in DistributedSampler()44 even_dist_); in Parse()
106 bool even_dist_; variable