mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
Summary: Weighted sampling reader dequeue randomly chooses a hive reader to read a mini-batch. This diff allows dequeue to output the index of the randomly chosen table to a specific blob. Reviewed By: kennyhorror Differential Revision: D6621070 fbshipit-source-id: 754b981fc2bcfdb0146d2a0a5b677e7cfe74211b |
||
|---|---|---|
| .. | ||
| blobs_queue.cc | ||
| blobs_queue.h | ||
| blobs_queue_db.cc | ||
| blobs_queue_db.h | ||
| CMakeLists.txt | ||
| queue_ops.cc | ||
| queue_ops.h | ||
| queue_ops_gpu.cc | ||
| rebatching_queue.cc | ||
| rebatching_queue.h | ||
| rebatching_queue_ops.cc | ||
| rebatching_queue_ops.h | ||