Searched refs:gpu_res1_half (Results 1 – 1 of 1) sorted by relevance
197 …Eigen::TensorMap<Eigen::Tensor<Eigen::half, 1>, Eigen::Aligned> gpu_res1_half(d_res1_half, num_ele… in test_cuda_trancendental() local211 gpu_res1_half.device(gpu_device) = gpu_float1.cast<Eigen::half>(); in test_cuda_trancendental()212 gpu_res1_half.device(gpu_device) = gpu_res1_half.exp(); in test_cuda_trancendental()