Searched refs:gpu_out_max (Results 1 – 1 of 1) sorted by relevance
50 …Eigen::TensorMap<Eigen::Tensor<DenseIndex, 1, Layout>, Aligned > gpu_out_max(d_out_max, Eigen::arr… in test_cuda_simple_argmax() local53 gpu_out_max.device(gpu_device) = gpu_in.argmax(); in test_cuda_simple_argmax()