Searched refs:gpu_kernel (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_cuda.cu | 345 Eigen::TensorMap<Eigen::Tensor<float, 1, DataLayout> > gpu_kernel(d_kernel, 4); in test_cuda_convolution_1d() local
349 gpu_out.device(gpu_device) = gpu_input.convolve(gpu_kernel, dims); in test_cuda_convolution_1d()
398 Eigen::TensorMap<Eigen::Tensor<float, 1, ColMajor> > gpu_kernel(d_kernel,4); in test_cuda_convolution_inner_dim_col_major_1d() local
402 gpu_out.device(gpu_device) = gpu_input.convolve(gpu_kernel, dims); in test_cuda_convolution_inner_dim_col_major_1d()
451 Eigen::TensorMap<Eigen::Tensor<float, 1, RowMajor> > gpu_kernel(d_kernel, 4); in test_cuda_convolution_inner_dim_row_major_1d() local
455 gpu_out.device(gpu_device) = gpu_input.convolve(gpu_kernel, dims); in test_cuda_convolution_inner_dim_row_major_1d()
505 Eigen::TensorMap<Eigen::Tensor<float, 2, DataLayout> > gpu_kernel(d_kernel,3,4); in test_cuda_convolution_2d() local
509 gpu_out.device(gpu_device) = gpu_input.convolve(gpu_kernel, dims); in test_cuda_convolution_2d()
569 Eigen::TensorMap<Eigen::Tensor<float, 3, DataLayout> > gpu_kernel(d_kernel,3,4,2); in test_cuda_convolution_3d() local
573 gpu_out.device(gpu_device) = gpu_input.convolve(gpu_kernel, dims); in test_cuda_convolution_3d()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | BUILD | 112 hdrs = if_gpu_is_configured(["gpu_kernel.h"]),
|