Home
last modified time | relevance | path

Searched refs:gpu_in1 (Results 1 – 6 of 6) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_sycl.cpp49 TensorMap<Tensor<float, 3>> gpu_in1(gpu_in1_data, tensorRange); in test_sycl_cpu() local
55 gpu_in1.device(sycl_device) = gpu_in1.constant(1.2f); in test_sycl_cpu()
67 gpu_out.device(sycl_device) = gpu_in1 * 1.2f; in test_sycl_cpu()
81 gpu_out.device(sycl_device) = gpu_in1 * gpu_in2; in test_sycl_cpu()
95 gpu_out.device(sycl_device) = gpu_in1 + gpu_in2; in test_sycl_cpu()
109 gpu_out.device(sycl_device) = gpu_in1 * gpu_in1; in test_sycl_cpu()
123 …gpu_out.device(sycl_device) = gpu_in1 * gpu_in1.constant(3.14f) + gpu_in2 * gpu_in2.constant(2.7f… in test_sycl_cpu()
138 gpu_out.device(sycl_device) =(gpu_in1 > gpu_in1.constant(0.5f)).select(gpu_in2, gpu_in3); in test_sycl_cpu()
Dcxx11_tensor_complex_cwise_ops_cuda.cu37 Eigen::TensorMap<Eigen::Tensor<std::complex<T>, 1, 0, int>, Eigen::Aligned> gpu_in1( in test_cuda_complex_cwise_ops() local
47 gpu_in1.device(gpu_device) = gpu_in1.constant(a); in test_cuda_complex_cwise_ops()
62 gpu_out.device(gpu_device) = gpu_in1 + gpu_in2; in test_cuda_complex_cwise_ops()
66 gpu_out.device(gpu_device) = gpu_in1 - gpu_in2; in test_cuda_complex_cwise_ops()
70 gpu_out.device(gpu_device) = gpu_in1 * gpu_in2; in test_cuda_complex_cwise_ops()
74 gpu_out.device(gpu_device) = gpu_in1 / gpu_in2; in test_cuda_complex_cwise_ops()
Dcxx11_tensor_forced_eval_sycl.cpp43 Eigen::TensorMap<Eigen::Tensor<float, 3>> gpu_in1(gpu_in1_data, tensorRange); in test_forced_eval_sycl() local
49 gpu_out.device(sycl_device) =(gpu_in1 + gpu_in2).eval() * gpu_in2; in test_forced_eval_sycl()
Dcxx11_tensor_complex_cuda.cu43 Eigen::TensorMap<Eigen::Tensor<std::complex<float>, 1, 0, int>, Eigen::Aligned> gpu_in1( in test_cuda_nullary() local
50 gpu_in1.device(gpu_device) = gpu_in1.constant(std::complex<float>(3.14f, 2.7f)); in test_cuda_nullary()
Dcxx11_tensor_cuda.cu41 Eigen::TensorMap<Eigen::Tensor<float, 1, 0, int>, Eigen::Aligned> gpu_in1( in test_cuda_nullary() local
46 gpu_in1.device(gpu_device) = gpu_in1.constant(3.14f); in test_cuda_nullary()
92 Eigen::TensorMap<Eigen::Tensor<float, 1>, Eigen::Aligned> gpu_in1( in test_cuda_elementwise_small() local
99 gpu_out.device(gpu_device) = gpu_in1 + gpu_in2; in test_cuda_elementwise_small()
147 …Eigen::TensorMap<Eigen::Tensor<float, 3> > gpu_in1(d_in1, Eigen::array<Eigen::DenseIndex, 3>(72,53… in test_cuda_elementwise() local
152 gpu_out.device(gpu_device) = gpu_in1 + gpu_in2 * gpu_in3; in test_cuda_elementwise()
189 Eigen::TensorMap<Eigen::Tensor<float, 1>, Eigen::Aligned> gpu_in1( in test_cuda_props() local
194 gpu_out.device(gpu_device) = (gpu_in1.isnan)(); in test_cuda_props()
227 Eigen::TensorMap<Eigen::Tensor<float, 4> > gpu_in1(d_in1, 72,53,97,113); in test_cuda_reduction() local
234 gpu_out.device(gpu_device) = gpu_in1.maximum(reduction_axis); in test_cuda_reduction()
Dcxx11_tensor_device.cu293 Eigen::TensorMap<Eigen::Tensor<float, 3> > gpu_in1(d_in1, 40,50,70); in test_gpu() local
297 GPUContext context(gpu_in1, gpu_in2, gpu_out); in test_gpu()