Searched refs:d_kernel (Results 1 – 1 of 1) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_cuda.cu | 332 float* d_kernel; in test_cuda_convolution_1d() local
335 cudaMalloc((void**)(&d_kernel), kernel_bytes); in test_cuda_convolution_1d()
339 cudaMemcpy(d_kernel, kernel.data(), kernel_bytes, cudaMemcpyHostToDevice); in test_cuda_convolution_1d()
345 Eigen::TensorMap<Eigen::Tensor<float, 1, DataLayout> > gpu_kernel(d_kernel, 4); in test_cuda_convolution_1d()
368 cudaFree(d_kernel); in test_cuda_convolution_1d()
385 float* d_kernel; in test_cuda_convolution_inner_dim_col_major_1d() local
388 cudaMalloc((void**)(&d_kernel), kernel_bytes); in test_cuda_convolution_inner_dim_col_major_1d()
392 cudaMemcpy(d_kernel, kernel.data(), kernel_bytes, cudaMemcpyHostToDevice); in test_cuda_convolution_inner_dim_col_major_1d()
398 Eigen::TensorMap<Eigen::Tensor<float, 1, ColMajor> > gpu_kernel(d_kernel,4); in test_cuda_convolution_inner_dim_col_major_1d()
421 cudaFree(d_kernel); in test_cuda_convolution_inner_dim_col_major_1d()
[all …]
|