H A D | conv_ops_gpu_3.cu.cc | 975 struct NHWCToNCHW<GPUDevice, T, NDIMS> { struct in namespace:tensorflow::functor 1048 template struct functor::NHWCToNCHW<GPUDevice, double, 4>; member in class:tensorflow::functor 1049 template struct functor::NHWCToNCHW<GPUDevice, float, 4>; member in class:tensorflow::functor 1050 template struct functor::NHWCToNCHW<GPUDevice, Eigen::half, 4>; member in class:tensorflow::functor 1067 template struct functor::NHWCToNCHW<GPUDevice, float, 5>; member in class:tensorflow::functor 1068 template struct functor::NHWCToNCHW<GPUDevice, Eigen::half, 5>; member in class:tensorflow::functor
|