Searched refs:kernel_bytes (Results 1 – 1 of 1) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_cuda.cu | 328 std::size_t kernel_bytes = kernel.size() * sizeof(float); 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()
381 std::size_t kernel_bytes = kernel.size() * sizeof(float); 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()
434 std::size_t kernel_bytes = kernel.size() * sizeof(float); in test_cuda_convolution_inner_dim_row_major_1d() local
441 cudaMalloc((void**)(&d_kernel), kernel_bytes); in test_cuda_convolution_inner_dim_row_major_1d()
445 cudaMemcpy(d_kernel, kernel.data(), kernel_bytes, cudaMemcpyHostToDevice); in test_cuda_convolution_inner_dim_row_major_1d()
488 std::size_t kernel_bytes = kernel.size() * sizeof(float); in test_cuda_convolution_2d() local
[all …]
|