Searched refs:local_sample_count (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torch/csrc/api/src/data/samplers/ |
| D | distributed.cpp | 55 size_t num_local_samples = local_sample_count(); in populate_indices()
142 begin_index_ = rank_ * local_sample_count(); in populate_indices()
143 end_index_ = begin_index_ + local_sample_count(); in populate_indices()
|
| /external/pytorch/torch/csrc/api/include/torch/data/samplers/ |
| D | distributed.h | 50 size_t local_sample_count() { in local_sample_count() function
|
| /external/pytorch/test/cpp/api/ |
| D | dataloader.cpp | 868 size_t local_sample_count, in TEST()
886 ASSERT_EQ(res.size(), local_sample_count * (i + 1)); in TEST()
893 size_t local_sample_count = in TEST() local
896 test_function(true, local_sample_count, output1, batch_size); in TEST()
898 local_sample_count = in TEST()
901 test_function(false, local_sample_count, output2, batch_size); in TEST()
964 size_t local_sample_count, in TEST()
983 ASSERT_EQ(res.size(), local_sample_count * (i + 1)); in TEST()
990 size_t local_sample_count = in TEST() local
993 test_function(true, local_sample_count, output1, batch_size); in TEST()
[all …]
|