Home | History | Annotate | Download | only in kernels

Lines Matching refs:GPUDevice

55 typedef Eigen::GpuDevice GPUDevice;
79 void RangeInit(const GPUDevice& d, const T start, const T delta,
89 void MoveValues(const GPUDevice& d, int32* keys, int32* values, int32* num_runs,
102 void CallGatherKernel(const GPUDevice& d, const T* params, const int32* indices,
327     const GPUDevice& device = c->eigen_device<GPUDevice>();
360     const GPUDevice& device = c->eigen_device<GPUDevice>();
378     functor::SetZeroFunctor<GPUDevice, int32> zero_functor;
440     const GPUDevice& device = c->eigen_device<GPUDevice>();