Searched refs:gpu_half (Results 1 – 2 of 2) sorted by relevance
38 Eigen::TensorMap<Eigen::Tensor<Eigen::half, 1>, Eigen::Aligned> gpu_half( in test_cuda_conversion() local45 gpu_half.device(gpu_device) = gpu_float.cast<Eigen::half>(); in test_cuda_conversion()46 gpu_conv.device(gpu_device) = gpu_half.cast<float>(); in test_cuda_conversion()
76 Eigen::TensorMap<Eigen::Tensor<Eigen::half, 1>, Eigen::Aligned> gpu_half( in test_cuda_conversion() local82 gpu_half.device(gpu_device) = gpu_float.cast<Eigen::half>(); in test_cuda_conversion()83 gpu_conv.device(gpu_device) = gpu_half.cast<float>(); in test_cuda_conversion()