H A D | conv_ops_gpu_3.cu.cc | 993 struct NCHWToNHWC<GPUDevice, T, NDIMS> { struct in namespace:tensorflow::functor 1052 template struct functor::NCHWToNHWC<GPUDevice, double, 4>; member in class:tensorflow::functor 1053 template struct functor::NCHWToNHWC<GPUDevice, float, 4>; member in class:tensorflow::functor 1054 template struct functor::NCHWToNHWC<GPUDevice, Eigen::half, 4>; member in class:tensorflow::functor 1070 template struct functor::NCHWToNHWC<GPUDevice, float, 5>; member in class:tensorflow::functor 1071 template struct functor::NCHWToNHWC<GPUDevice, Eigen::half, 5>; member in class:tensorflow::functor
|