Searched refs:SequentialSampler (Results 1 – 14 of 14) sorted by relevance
| /external/pytorch/torch/csrc/jit/mobile/train/ |
| D | sequential.cpp | 9 SequentialSampler::SequentialSampler(size_t size) : size_(size) {} in SequentialSampler() function in torch::jit::mobile::SequentialSampler
11 void SequentialSampler::reset(std::optional<size_t> new_size) { in reset()
18 optional<std::vector<size_t>> SequentialSampler::next(size_t batch_size) { in next()
30 void SequentialSampler::save(serialize::OutputArchive& archive) const { in save()
35 void SequentialSampler::load(serialize::InputArchive& archive) { in load()
40 size_t SequentialSampler::index() const noexcept { in index()
|
| D | sequential.h | 19 class TORCH_API SequentialSampler : public torch::data::samplers::Sampler<> {
23 explicit SequentialSampler(size_t size);
|
| /external/pytorch/torch/csrc/api/src/data/samplers/ |
| D | sequential.cpp | 12 SequentialSampler::SequentialSampler(size_t size) : size_(size) {} in SequentialSampler() function in torch::data::samplers::SequentialSampler
14 void SequentialSampler::reset(std::optional<size_t> new_size) { in reset()
21 std::optional<std::vector<size_t>> SequentialSampler::next(size_t batch_size) { in next()
33 void SequentialSampler::save(serialize::OutputArchive& archive) const { in save()
40 void SequentialSampler::load(serialize::InputArchive& archive) { in load()
49 size_t SequentialSampler::index() const noexcept { in index()
|
| /external/pytorch/test/cpp/api/ |
| D | dataloader.cpp | 99 samplers::SequentialSampler sampler(0); in TEST()
107 samplers::SequentialSampler, in TEST()
108 samplers::SequentialSampler>> in TEST()
111 samplers::SequentialSampler, in TEST()
112 samplers::SequentialSampler>>( in TEST()
270 samplers::SequentialSampler sampler(10); in TEST()
278 samplers::SequentialSampler sampler(5); in TEST()
285 samplers::SequentialSampler sampler(5); in TEST()
294 samplers::SequentialSampler sampler(5); in TEST()
308 samplers::SequentialSampler a(10); in TEST()
[all …]
|
| /external/pytorch/torch/csrc/api/include/torch/data/samplers/ |
| D | sequential.h | 22 class TORCH_API SequentialSampler : public Sampler<> {
26 explicit SequentialSampler(size_t size);
|
| /external/pytorch/torch/utils/data/datapipes/iter/ |
| D | combinatorics.py | 8 from torch.utils.data.sampler import Sampler, SequentialSampler
36 sampler: Type[Sampler] = SequentialSampler,
|
| /external/pytorch/torch/utils/data/ |
| D | __init__.py | 37 SequentialSampler,
|
| D | sampler.py | 112 class SequentialSampler(Sampler[int]): class
|
| D | dataloader.py | 35 SequentialSampler,
378 sampler = SequentialSampler(dataset) # type: ignore[arg-type]
|
| /external/pytorch/test/cpp/jit/ |
| D | test_lite_trainer.cpp | 328 TEST(LiteTrainerTest, SequentialSampler) { in TEST() argument
331 auto data_loader = torch::data::make_data_loader<mobile::SequentialSampler>( in TEST()
|
| /external/executorch/examples/qualcomm/scripts/ |
| D | mobilebert_fine_tune.py | 99 SequentialSampler,
180 sampler=SequentialSampler(dataset_val),
|
| /external/pytorch/test/quantization/core/experimental/ |
| D | quantization_util.py | 120 test_sampler = torch.utils.data.SequentialSampler(dataset_test)
|
| /external/pytorch/test/ |
| D | test_dataloader.py | 1508 sampler = torch.utils.data.SequentialSampler(self.dataset)
1567 dataset, sampler=torch.utils.data.SequentialSampler(dataset)
1580 torch.utils.data.SequentialSampler(dataset), 3, False
|
| /external/pytorch/docs/source/ |
| D | data.rst | 437 .. autoclass:: torch.utils.data.SequentialSampler
|