Searched refs:DistributedSequentialSampler (Results 1 – 3 of 3) sorted by relevance
97 DistributedSequentialSampler::DistributedSequentialSampler( in DistributedSequentialSampler() function in torch::data::samplers::DistributedSequentialSampler109 std::optional<std::vector<size_t>> DistributedSequentialSampler::next( in next()131 void DistributedSequentialSampler::reset(std::optional<size_t> new_size) { in reset()141 void DistributedSequentialSampler::populate_indices() { in populate_indices()147 void DistributedSequentialSampler::save( in save()155 void DistributedSequentialSampler::load(serialize::InputArchive& archive) { in load()161 size_t DistributedSequentialSampler::index() const noexcept { in index()
105 class TORCH_API DistributedSequentialSampler : public DistributedSampler<> {107 DistributedSequentialSampler(
943 samplers::DistributedSequentialSampler dss(sample_count); in TEST()967 std::vector<std::unique_ptr<samplers::DistributedSequentialSampler>> in TEST()972 std::make_unique<samplers::DistributedSequentialSampler>( in TEST()1004 samplers::DistributedSequentialSampler a(10); in TEST()1009 samplers::DistributedSequentialSampler b(10); in TEST()1014 samplers::DistributedSequentialSampler a(10); in TEST()1021 samplers::DistributedSequentialSampler b(10); in TEST()