Home
last modified time | relevance | path

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

/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_reduction_sycl.cpp30 array<IndexType, 2> tensorRange = {{num_rows, num_cols}}; in test_full_reductions_sum_sycl() local
112 array<IndexType, 2> tensorRange = {{num_rows, num_cols}}; in test_full_reductions_max_sycl() local
190 array<IndexType, 2> tensorRange = {{num_rows, num_cols}}; in test_full_reductions_mean_sycl() local
366 array<IndexType, 2> tensorRange = {{num_rows, num_cols}}; in test_full_reductions_min_sycl() local
447 array<IndexType, 3> tensorRange = {{dim_x, dim_y, dim_z}}; in test_first_dim_reductions_max_sycl() local
611 array<IndexType, 2> tensorRange = {{dim_x, dim_y}}; in test_first_dim_reductions_sum_sycl() local
655 array<IndexType, 3> tensorRange = {{dim_x, dim_y, dim_z}}; in test_first_dim_reductions_mean_sycl() local
701 array<IndexType, 3> tensorRange = {{dim_x, dim_y, dim_z}}; in test_last_dim_reductions_mean_sycl() local
746 array<IndexType, 3> tensorRange = {{dim_x, dim_y, dim_z}}; in test_last_dim_reductions_sum_sycl() local
787 auto tensorRange = Sizes<64, 32>(64, 32); in test_last_reductions_sum_sycl() local
[all …]
H A Dcxx11_tensor_sycl.cpp35 array<IndexType, 3> tensorRange = {{sizeDim1, sizeDim2, sizeDim3}}; in test_sycl_mem_transfers() local
72 array<IndexType, 1> tensorRange = {{size}}; in test_sycl_mem_sync() local
102 array<IndexType, 1> tensorRange = {{full_size}}; in test_sycl_mem_sync_offsets() local
156 array<IndexType, 1> tensorRange = {{full_size}}; in test_sycl_memset_offsets() local
185 array<IndexType, 3> tensorRange = {{sizeDim1, sizeDim2, sizeDim3}}; in test_sycl_computations() local
318 array<IndexType, 1> tensorRange = {{size}}; in test_sycl_cast() local
H A Dcxx11_tensor_generator_sycl.cpp38 array<IndexType, 1> tensorRange = {{sizeDim1}}; in test_1D_sycl() local
72 array<IndexType, 2> tensorRange = {{sizeDim1, sizeDim2}}; in test_2D_sycl() local
107 array<IndexType, 2> tensorRange = {{rows, cols}}; in test_gaussian_sycl() local
H A Dcxx11_tensor_chipping_sycl.cpp37 array<IndexType, 5> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4, sizeDim5}}; in test_static_chip_sycl() local
186 array<IndexType, 5> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4, sizeDim5}}; in test_dynamic_chip_sycl() local
334 array<IndexType, 5> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4, sizeDim5}}; in test_chip_in_expr() local
410 array<IndexType, 5> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4, sizeDim5}}; in test_chip_as_lvalue_sycl() local
H A Dcxx11_tensor_random_sycl.cpp33 array<IndexType, 2> tensorRange = {{sizeDim0, sizeDim1}}; in test_sycl_random_uniform() local
62 array<IndexType, 2> tensorRange = {{sizeDim0, sizeDim1}}; in test_sycl_random_normal() local
H A Dcxx11_tensor_math_sycl.cpp37 array<IndexType, 3> tensorRange = {{sizeDim1, sizeDim2, sizeDim3}}; in test_tanh_sycl() local
67 array<IndexType, 3> tensorRange = {{sizeDim1, sizeDim2, sizeDim3}}; in test_sigmoid_sycl() local
H A Dcxx11_tensor_padding_sycl.cpp40 array<IndexType, 4> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4}}; in test_simple_padding() local
96 array<IndexType, 4> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4}}; in test_padded_expr() local
H A Dcxx11_tensor_morphing_sycl.cpp145 array<IndexType, 5> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4, sizeDim5}}; in test_simple_slice() local
198 array<IndexType, 5> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4, sizeDim5}}; in test_strided_slice_as_rhs_sycl() local
266 array<IndexType, 2> tensorRange = {{sizeDim1, sizeDim2}}; in test_strided_slice_write_sycl() local
H A Dcxx11_tensor_reverse_sycl.cpp30 array<IndexType, 4> tensorRange = {{dim1, dim2, dim3, dim4}}; in test_simple_reverse() local
121 array<IndexType, 4> tensorRange = {{dim1, dim2, dim3, dim4}}; in test_expr_reverse() local
H A Dcxx11_tensor_device_sycl.cpp31 array<IndexType, 1> tensorRange = {{sizeDim1}}; in test_device_memory() local
H A Dcxx11_tensor_forced_eval_sycl.cpp30 Eigen::array<IndexType, 3> tensorRange = {{sizeDim1, sizeDim2, sizeDim3}}; in test_forced_eval_sycl() local
H A Dcxx11_tensor_shuffling_sycl.cpp35 array<IndexType, 4> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4}}; in test_simple_shuffling_sycl() local
H A Dcxx11_tensor_scan_sycl.cpp46 array<IndexType, 3> tensorRange = {{m_size, k_size, n_size}}; in test_sycl_cumsum() local
H A Dcxx11_tensor_inflation_sycl.cpp40 array<IndexType, 4> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4}}; in test_simple_inflation_sycl() local
H A Dcxx11_tensor_patch_sycl.cpp34 array<IndexType, 4> tensorRange = {{sizeDim1, sizeDim2, sizeDim3, sizeDim4}}; in test_simple_patch_sycl() local
H A Dcxx11_tensor_custom_op_sycl.cpp52 Eigen::array<IndexType, 2> tensorRange = {{sizeDim1, sizeDim2}}; in test_custom_unary_op_sycl() local
/aosp_15_r20/external/eigen/unsupported/doc/examples/SYCL/
H A DCwiseMul.cpp25 array<IndexType, 3> tensorRange = {{sizeDim1, sizeDim2, sizeDim3}}; in main() local