Searched refs:DistributedRandomSampler (Results 1 – 3 of 3) sorted by relevance
15 DistributedRandomSampler::DistributedRandomSampler( in DistributedRandomSampler() function in torch::data::samplers::DistributedRandomSampler28 std::optional<std::vector<size_t>> DistributedRandomSampler::next( in next()45 void DistributedRandomSampler::reset(std::optional<size_t> new_size) { in reset()54 void DistributedRandomSampler::populate_indices() { in populate_indices()70 void DistributedRandomSampler::save(serialize::OutputArchive& archive) const { in save()81 void DistributedRandomSampler::load(serialize::InputArchive& archive) { in load()93 size_t DistributedRandomSampler::index() const noexcept { in index()
72 class TORCH_API DistributedRandomSampler : public DistributedSampler<> {74 DistributedRandomSampler(
847 samplers::DistributedRandomSampler drs(sample_count); in TEST()871 std::vector<std::unique_ptr<samplers::DistributedRandomSampler>> samplers; in TEST()875 std::make_unique<samplers::DistributedRandomSampler>( in TEST()907 samplers::DistributedRandomSampler a(10); in TEST()912 samplers::DistributedRandomSampler b(10); in TEST()917 samplers::DistributedRandomSampler a(10); in TEST()924 samplers::DistributedRandomSampler b(10); in TEST()929 samplers::DistributedRandomSampler a(10); in TEST()934 samplers::DistributedRandomSampler b(10); in TEST()