Caffe2 - C++ API
A deep learning, cross platform ML framework
queue_ops_gpu.cc
1 
17 #include "caffe2/utils/math.h"
18 #include "queue_ops.h"
19 
20 #include "caffe2/core/context_gpu.h"
21 
22 namespace caffe2 {
23 
24 REGISTER_CUDA_OPERATOR(CreateBlobsQueue, CreateBlobsQueueOp<CUDAContext>);
25 REGISTER_CUDA_OPERATOR(EnqueueBlobs, EnqueueBlobsOp<CUDAContext>);
26 REGISTER_CUDA_OPERATOR(DequeueBlobs, DequeueBlobsOp<CUDAContext>);
27 REGISTER_CUDA_OPERATOR(CloseBlobsQueue, CloseBlobsQueueOp<CUDAContext>);
28 
29 REGISTER_CUDA_OPERATOR(SafeEnqueueBlobs, SafeEnqueueBlobsOp<CUDAContext>);
30 REGISTER_CUDA_OPERATOR(SafeDequeueBlobs, SafeDequeueBlobsOp<CUDAContext>);
31 
32 }
Copyright (c) 2016-present, Facebook, Inc.