pytorch/caffe2/queue
Lei Chen 8b5782ed5c Weighted sampling dequeue operator
Summary:
Similar to SafeDequeueBlobsOp, but add weight-based sampling for reading from multiple input BlobsQueue.

WeightedSampleDequeueBlobsOp will take a vector of weights (each weight is mapped to one input blob queue).
Based on probability, we will choose which BlobQueue to fetch.
WeightedSampleDequeueBlobsOp shall stop when any of input BlobQueue is empty.

Reviewed By: dzhulgakov

Differential Revision: D4905160

fbshipit-source-id: 5b1551e2250569f933a6c01ed04442843c5e0cb6
2017-04-19 12:02:06 -07:00
..
blobs_queue.h More perf stats for BlobsQueue 2017-03-24 14:03:28 -07:00
CMakeLists.txt Linter fix up to sync fbsource and github 2017-01-06 15:36:17 -08:00
queue_ops.cc Weighted sampling dequeue operator 2017-04-19 12:02:06 -07:00
queue_ops.h Weighted sampling dequeue operator 2017-04-19 12:02:06 -07:00
queue_ops_gpu.cc Add CUDA support for Safe{Enqueue,Dequeue}BlobsOps 2017-02-01 09:44:37 -08:00