Searched refs:d_out_min (Results 1 – 1 of 1) sorted by relevance
39 DenseIndex* d_out_min; in test_cuda_simple_argmax() local42 cudaMalloc((void**)(&d_out_min), out_bytes); in test_cuda_simple_argmax()51 …Eigen::TensorMap<Eigen::Tensor<DenseIndex, 1, Layout>, Aligned > gpu_out_min(d_out_min, Eigen::arr… in test_cuda_simple_argmax()57 …assert(cudaMemcpyAsync(out_min.data(), d_out_min, out_bytes, cudaMemcpyDeviceToHost, gpu_device.st… in test_cuda_simple_argmax()65 cudaFree(d_out_min); in test_cuda_simple_argmax()