Home
last modified time | relevance | path

Searched refs:tensorRange (Results 1 – 17 of 17) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_sycl.cpp35 array<IndexType, 3> tensorRange = {{sizeDim1, sizeDim2, sizeDim3}}; in test_sycl_mem_transfers() local
36 Tensor<DataType, 3, DataLayout, IndexType> in1(tensorRange); in test_sycl_mem_transfers()
37 Tensor<DataType, 3, DataLayout, IndexType> out1(tensorRange); in test_sycl_mem_transfers()
38 Tensor<DataType, 3, DataLayout, IndexType> out2(tensorRange); in test_sycl_mem_transfers()
39 Tensor<DataType, 3, DataLayout, IndexType> out3(tensorRange); in test_sycl_mem_transfers()
46 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu1(gpu_data1, tensorRange); in test_sycl_mem_transfers()
47 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu2(gpu_data2, tensorRange); in test_sycl_mem_transfers()
72 array<IndexType, 1> tensorRange = {{size}}; in test_sycl_mem_sync() local
73 Tensor<DataType, 1, DataLayout, IndexType> in1(tensorRange); in test_sycl_mem_sync()
74 Tensor<DataType, 1, DataLayout, IndexType> in2(tensorRange); in test_sycl_mem_sync()
[all …]
Dcxx11_tensor_generator_sycl.cpp38 array<IndexType, 1> tensorRange = {{sizeDim1}}; in test_1D_sycl() local
39 Tensor<DataType, 1, DataLayout,IndexType> vec(tensorRange); in test_1D_sycl()
40 Tensor<DataType, 1, DataLayout,IndexType> result(tensorRange); in test_1D_sycl()
46 TensorMap<Tensor<DataType, 1, DataLayout,IndexType>> gpu_vec(gpu_data_vec, tensorRange); in test_1D_sycl()
47 TensorMap<Tensor<DataType, 1, DataLayout,IndexType>> gpu_result(gpu_data_result, tensorRange); in test_1D_sycl()
72 array<IndexType, 2> tensorRange = {{sizeDim1, sizeDim2}}; in test_2D_sycl() local
73 Tensor<DataType, 2, DataLayout,IndexType> matrix(tensorRange); in test_2D_sycl()
74 Tensor<DataType, 2, DataLayout,IndexType> result(tensorRange); in test_2D_sycl()
80 TensorMap<Tensor<DataType, 2, DataLayout,IndexType>> gpu_matrix(gpu_data_matrix, tensorRange); in test_2D_sycl()
81 TensorMap<Tensor<DataType, 2, DataLayout,IndexType>> gpu_result(gpu_data_result, tensorRange); in test_2D_sycl()
[all …]
Dcxx11_tensor_math_sycl.cpp37 array<IndexType, 3> tensorRange = {{sizeDim1, sizeDim2, sizeDim3}}; in test_tanh_sycl() local
38 Tensor<DataType, 3, DataLayout, IndexType> in(tensorRange); in test_tanh_sycl()
39 Tensor<DataType, 3, DataLayout, IndexType> out(tensorRange); in test_tanh_sycl()
40 Tensor<DataType, 3, DataLayout, IndexType> out_cpu(tensorRange); in test_tanh_sycl()
47 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu1(gpu_data1, tensorRange); in test_tanh_sycl()
48 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu2(gpu_data2, tensorRange); in test_tanh_sycl()
67 array<IndexType, 3> tensorRange = {{sizeDim1, sizeDim2, sizeDim3}}; in test_sigmoid_sycl() local
68 Tensor<DataType, 3, DataLayout, IndexType> in(tensorRange); in test_sigmoid_sycl()
69 Tensor<DataType, 3, DataLayout, IndexType> out(tensorRange); in test_sigmoid_sycl()
70 Tensor<DataType, 3, DataLayout, IndexType> out_cpu(tensorRange); in test_sigmoid_sycl()
[all …]
Dcxx11_tensor_forced_eval_sycl.cpp30 Eigen::array<IndexType, 3> tensorRange = {{sizeDim1, sizeDim2, sizeDim3}}; in test_forced_eval_sycl() local
31 Eigen::Tensor<DataType, 3, DataLayout, IndexType> in1(tensorRange); in test_forced_eval_sycl()
32 Eigen::Tensor<DataType, 3, DataLayout, IndexType> in2(tensorRange); in test_forced_eval_sycl()
33 Eigen::Tensor<DataType, 3, DataLayout, IndexType> out(tensorRange); in test_forced_eval_sycl()
43 …n::TensorMap<Eigen::Tensor<DataType, 3, DataLayout, IndexType>> gpu_in1(gpu_in1_data, tensorRange); in test_forced_eval_sycl()
44 …n::TensorMap<Eigen::Tensor<DataType, 3, DataLayout, IndexType>> gpu_in2(gpu_in2_data, tensorRange); in test_forced_eval_sycl()
45 …n::TensorMap<Eigen::Tensor<DataType, 3, DataLayout, IndexType>> gpu_out(gpu_out_data, tensorRange); in test_forced_eval_sycl()
Dcxx11_tensor_reverse_sycl.cpp30 array<IndexType, 4> tensorRange = {{dim1, dim2, dim3, dim4}}; in test_simple_reverse() local
31 Tensor<DataType, 4, DataLayout, IndexType> tensor(tensorRange); in test_simple_reverse()
32 Tensor<DataType, 4, DataLayout, IndexType> reversed_tensor(tensorRange); in test_simple_reverse()
47 tensorRange); in test_simple_reverse()
49 tensorRange); in test_simple_reverse()
121 array<IndexType, 4> tensorRange = {{dim1, dim2, dim3, dim4}}; in test_expr_reverse() local
122 Tensor<DataType, 4, DataLayout, IndexType> tensor(tensorRange); in test_expr_reverse()
123 Tensor<DataType, 4, DataLayout, IndexType> expected(tensorRange); in test_expr_reverse()
124 Tensor<DataType, 4, DataLayout, IndexType> result(tensorRange); in test_expr_reverse()
141 tensorRange); in test_expr_reverse()
[all …]
Dcxx11_tensor_reduction_sycl.cpp30 array<IndexType, 2> tensorRange = {{num_rows, num_cols}}; in test_full_reductions_sum_sycl() local
34 Tensor<DataType, 2, DataLayout, IndexType> in(tensorRange); in test_full_reductions_sum_sycl()
48 tensorRange); in test_full_reductions_sum_sycl()
112 array<IndexType, 2> tensorRange = {{num_rows, num_cols}}; in test_full_reductions_max_sycl() local
114 Tensor<DataType, 2, DataLayout, IndexType> in(tensorRange); in test_full_reductions_max_sycl()
127 tensorRange); in test_full_reductions_max_sycl()
190 array<IndexType, 2> tensorRange = {{num_rows, num_cols}}; in test_full_reductions_mean_sycl() local
195 Tensor<DataType, 2, DataLayout, IndexType> in(tensorRange); in test_full_reductions_mean_sycl()
196 Tensor<DataType, 2, DataLayout, IndexType> in_arg1(tensorRange); in test_full_reductions_mean_sycl()
197 Tensor<DataType, 2, DataLayout, IndexType> in_arg2(tensorRange); in test_full_reductions_mean_sycl()
[all …]
Dcxx11_tensor_shuffling_sycl.cpp35 array<IndexType, 4> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4}}; in test_simple_shuffling_sycl() local
36 Tensor<DataType, 4, DataLayout, IndexType> tensor(tensorRange); in test_simple_shuffling_sycl()
37 Tensor<DataType, 4, DataLayout, IndexType> no_shuffle(tensorRange); in test_simple_shuffling_sycl()
50 tensorRange); in test_simple_shuffling_sycl()
52 tensorRange); in test_simple_shuffling_sycl()
Dcxx11_tensor_padding_sycl.cpp40 array<IndexType, 4> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4}}; in test_simple_padding() local
42 Tensor<DataType, 4, DataLayout, IndexType> tensor(tensorRange); in test_simple_padding()
62 TensorMap<Tensor<DataType, 4,DataLayout,IndexType>> gpu1(gpu_data1, tensorRange); in test_simple_padding()
96 array<IndexType, 4> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4}}; in test_padded_expr() local
98 Tensor<DataType, 4, DataLayout, IndexType> tensor(tensorRange); in test_padded_expr()
116 TensorMap<Tensor<DataType, 4,DataLayout,IndexType>> gpu1(gpu_data1, tensorRange); in test_padded_expr()
Dcxx11_tensor_inflation_sycl.cpp40 array<IndexType, 4> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4}}; in test_simple_inflation_sycl() local
41 Tensor<DataType, 4, DataLayout,IndexType> tensor(tensorRange); in test_simple_inflation_sycl()
42 Tensor<DataType, 4, DataLayout,IndexType> no_stride(tensorRange); in test_simple_inflation_sycl()
56 TensorMap<Tensor<DataType, 4, DataLayout,IndexType>> gpu_tensor(gpu_data_tensor, tensorRange); in test_simple_inflation_sycl()
57 …ensorMap<Tensor<DataType, 4, DataLayout,IndexType>> gpu_no_stride(gpu_data_no_stride, tensorRange); in test_simple_inflation_sycl()
Dcxx11_tensor_random_sycl.cpp33 array<IndexType, 2> tensorRange = {{sizeDim0, sizeDim1}}; in test_sycl_random_uniform() local
36 TensorMap<Tensor<DataType, 2, DataLayout, IndexType>> gpu_out(d_out, tensorRange); in test_sycl_random_uniform()
62 array<IndexType, 2> tensorRange = {{sizeDim0, sizeDim1}}; in test_sycl_random_normal() local
65 TensorMap<Tensor<DataType, 2, DataLayout, IndexType>> gpu_out(d_out, tensorRange); in test_sycl_random_normal()
Dcxx11_tensor_chipping_sycl.cpp37 array<IndexType, 5> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4, sizeDim5}}; in test_static_chip_sycl() local
40 Tensor<DataType, 5, DataLayout,IndexType> tensor(tensorRange); in test_static_chip_sycl()
50 TensorMap<Tensor<DataType, 5, DataLayout,IndexType>> gpu_tensor(gpu_data_tensor, tensorRange); in test_static_chip_sycl()
186 array<IndexType, 5> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4, sizeDim5}}; in test_dynamic_chip_sycl() local
189 Tensor<DataType, 5, DataLayout,IndexType> tensor(tensorRange); in test_dynamic_chip_sycl()
199 TensorMap<Tensor<DataType, 5, DataLayout,IndexType>> gpu_tensor(gpu_data_tensor, tensorRange); in test_dynamic_chip_sycl()
334 array<IndexType, 5> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4, sizeDim5}}; in test_chip_in_expr() local
337 Tensor<DataType, 5, DataLayout,IndexType> tensor(tensorRange); in test_chip_in_expr()
350 TensorMap<Tensor<DataType, 5, DataLayout,IndexType>> gpu_tensor(gpu_data_tensor, tensorRange); in test_chip_in_expr()
410 array<IndexType, 5> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4, sizeDim5}}; in test_chip_as_lvalue_sycl() local
[all …]
Dcxx11_tensor_device_sycl.cpp31 array<IndexType, 1> tensorRange = {{sizeDim1}}; in test_device_memory() local
32 Tensor<DataType, 1, DataLayout,IndexType> in(tensorRange); in test_device_memory()
33 Tensor<DataType, 1, DataLayout,IndexType> in1(tensorRange); in test_device_memory()
Dcxx11_tensor_morphing_sycl.cpp145 array<IndexType, 5> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4, sizeDim5}}; in test_simple_slice() local
146 Tensor<DataType, 5,DataLayout, IndexType> tensor(tensorRange); in test_simple_slice()
153 TensorMap<Tensor<DataType, 5,DataLayout, IndexType>> gpu1(gpu_data1, tensorRange); in test_simple_slice()
198 array<IndexType, 5> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4, sizeDim5}}; in test_strided_slice_as_rhs_sycl() local
199 Tensor<DataType, 5, DataLayout, IndexType> tensor(tensorRange); in test_strided_slice_as_rhs_sycl()
210 TensorMap<Tensor<DataType, 5,DataLayout, IndexType>> gpu1(gpu_data1, tensorRange); in test_strided_slice_as_rhs_sycl()
266 array<IndexType, 2> tensorRange = {{sizeDim1, sizeDim2}}; in test_strided_slice_write_sycl() local
267 Tensor<DataType, 2, DataLayout, IndexType> tensor(tensorRange),tensor2(tensorRange); in test_strided_slice_write_sycl()
280 TensorMap<Tensor<DataType, 2,DataLayout,IndexType>> gpu1(gpu_data1, tensorRange); in test_strided_slice_write_sycl()
281 TensorMap<Tensor<DataType, 2,DataLayout,IndexType>> gpu2(gpu_data2, tensorRange); in test_strided_slice_write_sycl()
Dcxx11_tensor_scan_sycl.cpp46 array<IndexType, 3> tensorRange = {{m_size, k_size, n_size}}; in test_sycl_cumsum() local
48 gpu_data_in, tensorRange); in test_sycl_cumsum()
50 gpu_data_out, tensorRange); in test_sycl_cumsum()
Dcxx11_tensor_custom_op_sycl.cpp52 Eigen::array<IndexType, 2> tensorRange = {{sizeDim1, sizeDim2}}; in test_custom_unary_op_sycl() local
55 Eigen::Tensor<DataType, 2, DataLayout, IndexType> in1(tensorRange); in test_custom_unary_op_sycl()
62 TensorType gpu_in1(gpu_in1_data, tensorRange); in test_custom_unary_op_sycl()
Dcxx11_tensor_patch_sycl.cpp34 array<IndexType, 4> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4}}; in test_simple_patch_sycl() local
42 Tensor<DataType, 4, DataLayout,IndexType> tensor(tensorRange); in test_simple_patch_sycl()
58 TensorMap<Tensor<DataType, 4, DataLayout,IndexType>> gpu_tensor(gpu_data_tensor, tensorRange); in test_simple_patch_sycl()
/external/eigen/unsupported/doc/examples/SYCL/
DCwiseMul.cpp25 array<IndexType, 3> tensorRange = {{sizeDim1, sizeDim2, sizeDim3}}; in main() local
28 Tensor<DataType, 3,DataLayout, IndexType> in1(tensorRange); in main()
29 Tensor<DataType, 3,DataLayout, IndexType> in2(tensorRange); in main()
30 Tensor<DataType, 3,DataLayout, IndexType> out(tensorRange); in main()
42 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu_in1(gpu_in1_data, tensorRange); in main()
43 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu_in2(gpu_in2_data, tensorRange); in main()
44 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu_out(gpu_out_data, tensorRange); in main()