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 …]
|