Home | History | Annotate | Download | only in kernels

Lines Matching refs:GPUDevice

79 using GPUDevice = Eigen::GpuDevice;
87 extern template Status RingAllreduce<GPUDevice, int>(OpKernelContext*,
90 extern template Status RingAllreduce<GPUDevice, long long>(OpKernelContext*,
93 extern template Status RingAllreduce<GPUDevice, float>(OpKernelContext*,
96 extern template Status RingAllgather<GPUDevice, int>(OpKernelContext*,
100 extern template Status RingAllgather<GPUDevice, long long>(
102 extern template Status RingAllgather<GPUDevice, float>(
124 // Return true if the templated type is GPUDevice, otherwise false.
128 bool IsGPUDevice<GPUDevice>() {
459   // Use CPUDevice instead of GPUDevice if no CUDA, to ensure we don't
462 #define GPU_DEVICE_IF_CUDA GPUDevice
532 //      (inaccessible) GPU stream separate from the TF GPUDevice streams, we
876                         MPIInitOp<GPUDevice>);
902                         MPISizeOp<GPUDevice>);
928                         MPIRankOp<GPUDevice>);
955     MPILocalRankOp<GPUDevice>);
1029                         MPIAllreduceOp<GPUDevice>);
1125     MPIAllgatherOp<GPUDevice>);