Searched refs:d_out_max (Results 1 – 1 of 1) sorted by relevance
38 DenseIndex* d_out_max; in test_cuda_simple_argmax() local41 cudaMalloc((void**)(&d_out_max), out_bytes); in test_cuda_simple_argmax()50 …Eigen::TensorMap<Eigen::Tensor<DenseIndex, 1, Layout>, Aligned > gpu_out_max(d_out_max, Eigen::arr… in test_cuda_simple_argmax()56 …assert(cudaMemcpyAsync(out_max.data(), d_out_max, out_bytes, cudaMemcpyDeviceToHost, gpu_device.st… in test_cuda_simple_argmax()64 cudaFree(d_out_max); in test_cuda_simple_argmax()