| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_complex_cwise_ops_gpu.cu | 36 Eigen::TensorMap<Eigen::Tensor<std::complex<T>, 1, 0, int>, Eigen::Aligned> gpu_in2( in test_cuda_complex_cwise_ops() local
45 gpu_in2.device(gpu_device) = gpu_in2.constant(b); in test_cuda_complex_cwise_ops()
61 gpu_out.device(gpu_device) = gpu_in1 + gpu_in2; in test_cuda_complex_cwise_ops()
65 gpu_out.device(gpu_device) = gpu_in1 - gpu_in2; in test_cuda_complex_cwise_ops()
69 gpu_out.device(gpu_device) = gpu_in1 * gpu_in2; in test_cuda_complex_cwise_ops()
73 gpu_out.device(gpu_device) = gpu_in1 / gpu_in2; in test_cuda_complex_cwise_ops()
|
| D | cxx11_tensor_concatenation_sycl.cpp | 51 …Eigen::TensorMap<Eigen::Tensor<DataType, 3, DataLayout, IndexType>> gpu_in2(gpu_in2_data, rightRan… in test_simple_concatenation() local
60 gpu_out1.device(sycl_device) =gpu_in1.concatenate(gpu_in2, 0); in test_simple_concatenation()
79 gpu_out2.device(sycl_device) =gpu_in1.concatenate(gpu_in2, 1); in test_simple_concatenation()
98 gpu_out3.device(sycl_device) =gpu_in1.concatenate(gpu_in2, 2); in test_simple_concatenation()
145 …Eigen::TensorMap<Eigen::Tensor<DataType, 2, DataLayout, IndexType>> gpu_in2(gpu_in2_data, rightRan… in test_concatenation_as_lvalue() local
153 gpu_in1.concatenate(gpu_in2, 0).device(sycl_device) =gpu_out; in test_concatenation_as_lvalue()
|
| D | cxx11_tensor_forced_eval_sycl.cpp | 44 …Eigen::TensorMap<Eigen::Tensor<DataType, 3, DataLayout, IndexType>> gpu_in2(gpu_in2_data, tensorRa… in test_forced_eval_sycl() local
49 gpu_out.device(sycl_device) =(gpu_in1 + gpu_in2).eval() * gpu_in2; in test_forced_eval_sycl()
|
| D | cxx11_tensor_sycl.cpp | 200 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu_in2(gpu_in2_data, tensorRange); in test_sycl_computations() local
235 gpu_out.device(sycl_device) = gpu_in1 * gpu_in2; in test_sycl_computations()
251 gpu_out.device(sycl_device) = gpu_in1 + gpu_in2; in test_sycl_computations()
281 …gpu_out.device(sycl_device) = gpu_in1 * gpu_in1.constant(3.14f) + gpu_in2 * gpu_in2.constant(2.7f… in test_sycl_computations()
297 gpu_out.device(sycl_device) =(gpu_in1 > gpu_in1.constant(0.5f)).select(gpu_in2, gpu_in3); in test_sycl_computations()
|
| D | cxx11_tensor_custom_op_sycl.cpp | 128 TensorType gpu_in2(gpu_in2_data, tensorRange2); in test_custom_binary_op_sycl() local
137 gpu_out.device(sycl_device) = gpu_in1.customOp(gpu_in2, BatchMatMul<TensorType>()); in test_custom_binary_op_sycl()
|
| D | cxx11_tensor_complex_gpu.cu | 42 Eigen::TensorMap<Eigen::Tensor<std::complex<float>, 1, 0, int>, Eigen::Aligned> gpu_in2( in test_cuda_nullary() local
48 gpu_out2.device(gpu_device) = gpu_in2.abs(); in test_cuda_nullary()
|
| D | cxx11_tensor_gpu.cu | 44 Eigen::TensorMap<Eigen::Tensor<float, 1, 0, int>, Eigen::Aligned> gpu_in2( in test_gpu_nullary() local
48 gpu_in2.device(gpu_device) = gpu_in2.random(); in test_gpu_nullary()
95 Eigen::TensorMap<Eigen::Tensor<float, 1>, Eigen::Aligned> gpu_in2( in test_gpu_elementwise_small() local
100 gpu_out.device(gpu_device) = gpu_in1 + gpu_in2; in test_gpu_elementwise_small()
149 …Eigen::TensorMap<Eigen::Tensor<float, 3> > gpu_in2(d_in2, Eigen::array<Eigen::DenseIndex, 3>(72,53… in test_gpu_elementwise() local
153 gpu_out.device(gpu_device) = gpu_in1 + gpu_in2 * gpu_in3; in test_gpu_elementwise()
|
| D | cxx11_tensor_device.cu | 293 Eigen::TensorMap<Eigen::Tensor<float, 3> > gpu_in2(d_in2, 40,50,70); in test_gpu() local
296 GPUContext context(gpu_in1, gpu_in2, gpu_out); in test_gpu()
|
| /external/eigen/unsupported/doc/examples/SYCL/ |
| D | CwiseMul.cpp | 43 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu_in2(gpu_in2_data, tensorRange); in main() local
49 gpu_out.device(sycl_device) = gpu_in1 * gpu_in2; in main()
|