Home
last modified time | relevance | path

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

/external/eigen/unsupported/test/
Dcxx11_tensor_complex_cwise_ops_cuda.cu39 Eigen::TensorMap<Eigen::Tensor<std::complex<T>, 1, 0, int>, Eigen::Aligned> gpu_in2( in test_cuda_complex_cwise_ops() local
48 gpu_in2.device(gpu_device) = gpu_in2.constant(b); 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_sycl.cpp50 TensorMap<Tensor<float, 3>> gpu_in2(gpu_in2_data, tensorRange); in test_sycl_cpu() local
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()
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_forced_eval_sycl.cpp44 Eigen::TensorMap<Eigen::Tensor<float, 3>> gpu_in2(gpu_in2_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.cu45 Eigen::TensorMap<Eigen::Tensor<std::complex<float>, 1, 0, int>, Eigen::Aligned> gpu_in2( in test_cuda_nullary() local
51 gpu_out2.device(gpu_device) = gpu_in2.abs(); in test_cuda_nullary()
Dcxx11_tensor_cuda.cu43 Eigen::TensorMap<Eigen::Tensor<float, 1, 0, int>, Eigen::Aligned> gpu_in2( in test_cuda_nullary() local
47 gpu_in2.device(gpu_device) = gpu_in2.random(); in test_cuda_nullary()
94 Eigen::TensorMap<Eigen::Tensor<float, 1>, Eigen::Aligned> gpu_in2( in test_cuda_elementwise_small() local
99 gpu_out.device(gpu_device) = gpu_in1 + gpu_in2; in test_cuda_elementwise_small()
148 …Eigen::TensorMap<Eigen::Tensor<float, 3> > gpu_in2(d_in2, 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()
Dcxx11_tensor_device.cu294 Eigen::TensorMap<Eigen::Tensor<float, 3> > gpu_in2(d_in2, 40,50,70); in test_gpu() local
297 GPUContext context(gpu_in1, gpu_in2, gpu_out); in test_gpu()