| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_morphing_sycl.cpp | 51 TensorMap<Tensor<DataType, 5,DataLayout, IndexType>> gpu1(gpu_data1, dim1); in test_simple_reshape() local
105 TensorMap< Tensor<DataType, 3, DataLayout, IndexType> > gpu1(gpu_data1, dim1); in test_reshape_as_lvalue() local
153 TensorMap<Tensor<DataType, 5,DataLayout, IndexType>> gpu1(gpu_data1, tensorRange); in test_simple_slice() local
210 TensorMap<Tensor<DataType, 5,DataLayout, IndexType>> gpu1(gpu_data1, tensorRange); in test_strided_slice_as_rhs_sycl() local
280 TensorMap<Tensor<DataType, 2,DataLayout,IndexType>> gpu1(gpu_data1, tensorRange); in test_strided_slice_write_sycl() local
|
| H A D | cxx11_tensor_sycl.cpp | 46 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu1(gpu_data1, tensorRange); in test_sycl_mem_transfers() local
82 TensorMap<Tensor<DataType, 1, DataLayout, IndexType>> gpu1(gpu_data, tensorRange); in test_sycl_mem_sync() local
107 TensorMap<tensor_type> gpu1(gpu_data, tensorRange); in test_sycl_mem_sync_offsets() local
166 TensorMap<tensor_type> gpu1(gpu_data, tensorRange); in test_sycl_memset_offsets() local
|
| H A D | cxx11_tensor_math_sycl.cpp | 47 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu1(gpu_data1, tensorRange); in test_tanh_sycl() local
77 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu1(gpu_data1, tensorRange); in test_sigmoid_sycl() local
|
| H A D | cxx11_tensor_layout_swap_sycl.cpp | 43 TensorMap<Tensor<DataType, 3, ColMajor, IndexType>> gpu1(gpu_data1, tensorColRange); in test_simple_swap_sycl() local
86 TensorMap<Tensor<DataType, 3, ColMajor, IndexType>> gpu1(gpu_data1, tensorColRange); in test_swap_as_lvalue_sycl() local
|
| H A D | cxx11_tensor_padding_sycl.cpp | 62 TensorMap<Tensor<DataType, 4,DataLayout,IndexType>> gpu1(gpu_data1, tensorRange); in test_simple_padding() local
116 TensorMap<Tensor<DataType, 4,DataLayout,IndexType>> gpu1(gpu_data1, tensorRange); in test_padded_expr() local
|
| H A D | cxx11_tensor_shuffling_sycl.cpp | 49 TensorMap<Tensor<DataType, 4, DataLayout, IndexType>> gpu1(gpu_data1, in test_simple_shuffling_sycl() local
|
| H A D | cxx11_tensor_image_op_sycl.cpp | 65 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu1(gpu_data1, input_range); in test_image_op_sycl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | device_util_test.cc | 72 const std::string gpu1 = "/job:worker/replica:1/task:2/device:GPU:1"; in TEST() local
187 const std::string gpu1 = "/job:worker/replica:0/task:0/device:GPU:1"; in TEST() local
|