Searched refs:tensorRange (Results 1 - 3 of 3) sorted by relevance
/external/eigen/unsupported/test/ |
H A D | cxx11_tensor_forced_eval_sycl.cpp | 30 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 D | cxx11_tensor_reduction_sycl.cpp | 29 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 D | cxx11_tensor_sycl.cpp | 35 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