Searched refs:tensorRange (Results 1 - 3 of 3) sorted by relevance

/external/eigen/unsupported/test/
H A Dcxx11_tensor_forced_eval_sycl.cpp30 Eigen::array<int, 3> tensorRange = {{sizeDim1, sizeDim2, sizeDim3}}; local
31 Eigen::Tensor<float, 3> in1(tensorRange);
32 Eigen::Tensor<float, 3> in2(tensorRange);
33 Eigen::Tensor<float, 3> out(tensorRange);
43 Eigen::TensorMap<Eigen::Tensor<float, 3>> gpu_in1(gpu_in1_data, tensorRange);
44 Eigen::TensorMap<Eigen::Tensor<float, 3>> gpu_in2(gpu_in2_data, tensorRange);
45 Eigen::TensorMap<Eigen::Tensor<float, 3>> gpu_out(gpu_out_data, tensorRange);
H A Dcxx11_tensor_reduction_sycl.cpp29 array<int, 2> tensorRange = {{num_rows, num_cols}}; local
31 Tensor<float, 2> in(tensorRange);
42 TensorMap<Tensor<float, 2> > in_gpu(gpu_in_data, tensorRange);
61 array<int, 3> tensorRange = {{dim_x, dim_y, dim_z}}; local
66 Tensor<float, 3> in(tensorRange);
77 TensorMap<Tensor<float, 3> > in_gpu(gpu_in_data, tensorRange);
99 array<int, 3> tensorRange = {{dim_x, dim_y, dim_z}}; local
104 Tensor<float, 3> in(tensorRange);
115 TensorMap<Tensor<float, 3> > in_gpu(gpu_in_data, tensorRange);
H A Dcxx11_tensor_sycl.cpp35 array<int, 3> tensorRange = {{sizeDim1, sizeDim2, sizeDim3}}; local
36 Tensor<float, 3> in1(tensorRange);
37 Tensor<float, 3> in2(tensorRange);
38 Tensor<float, 3> in3(tensorRange);
39 Tensor<float, 3> out(tensorRange);
49 TensorMap<Tensor<float, 3>> gpu_in1(gpu_in1_data, tensorRange);
50 TensorMap<Tensor<float, 3>> gpu_in2(gpu_in2_data, tensorRange);
51 TensorMap<Tensor<float, 3>> gpu_in3(gpu_in3_data, tensorRange);
52 TensorMap<Tensor<float, 3>> gpu_out(gpu_out_data, tensorRange);

Completed in 84 milliseconds