Home
last modified time | relevance | path

Searched refs:tensor_range (Results 1 – 3 of 3) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_builtins_sycl.cpp54 const array<int64_t, 3>& tensor_range) { in test_unary_builtins_for_scalar() argument
59 Tensor<DataType, 3, DataLayout, int64_t> in(tensor_range); in test_unary_builtins_for_scalar()
60 Tensor<DataType, 3, DataLayout, int64_t> out(tensor_range); in test_unary_builtins_for_scalar()
68 TensorMap<Tensor<DataType, 3, DataLayout, int64_t>> gpu(gpu_data, tensor_range); in test_unary_builtins_for_scalar()
69 TensorMap<Tensor<DataType, 3, DataLayout, int64_t>> gpu_out(gpu_data_out, tensor_range); in test_unary_builtins_for_scalar()
88 Tensor<DataType, 3, DataLayout, int64_t> out(tensor_range); in test_unary_builtins_for_scalar()
93 TensorMap<Tensor<DataType, 3, DataLayout, int64_t>> gpu_out(gpu_data_out, tensor_range); in test_unary_builtins_for_scalar()
142 const array<int64_t, 3>& tensor_range) { in DECLARE_UNARY_STRUCT()
145 op_##FUNC>(sycl_device, tensor_range) in DECLARE_UNARY_STRUCT()
165 const array<int64_t, 3>& tensor_range) { in test_unary_builtins_return_bool() argument
[all …]
Dcxx11_tensor_reduction_sycl.cpp72 array<IndexType, 2> tensor_range = {{num_rows, num_cols}}; in test_full_reductions_sum_with_offset_sycl() local
73 const IndexType n_elems = internal::array_prod(tensor_range); in test_full_reductions_sum_with_offset_sycl()
75 data_tensor in(tensor_range); in test_full_reductions_sum_with_offset_sycl()
80 array<IndexType, 2> tensor_offset_range(tensor_range); in test_full_reductions_sum_with_offset_sycl()
146 array<IndexType, 2> tensor_range = {{num_rows, num_cols}}; in test_full_reductions_max_with_offset_sycl() local
147 const IndexType n_elems = internal::array_prod(tensor_range); in test_full_reductions_max_with_offset_sycl()
149 data_tensor in(tensor_range); in test_full_reductions_max_with_offset_sycl()
154 array<IndexType, 2> tensor_offset_range(tensor_range); in test_full_reductions_max_with_offset_sycl()
279 array<IndexType, 2> tensor_range = {{num_rows, num_cols}}; in test_full_reductions_mean_with_offset_sycl() local
280 const IndexType n_elems = internal::array_prod(tensor_range); in test_full_reductions_mean_with_offset_sycl()
[all …]
Dcxx11_tensor_morphing_sycl.cpp322 Eigen::array<IndexType, 5> tensor_range{{4, 1, 1, 1, 6}}; in run_eigen() local
325 TensorI64 out_tensor_gpu(tensor_range); in run_eigen()
326 TensorI64 out_tensor_cpu(tensor_range); in run_eigen()
334 TensorMI64 out_gpu(out_gpu_data, tensor_range); in run_eigen()