| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_morphing_sycl.cpp | 51 TensorMap<Tensor<DataType, 5,DataLayout, IndexType>> gpu1(gpu_data1, dim1); in test_simple_reshape() local
58 gpu2.device(sycl_device)=gpu1.reshape(dim2); in test_simple_reshape()
61 gpu3.device(sycl_device)=gpu1.reshape(dim3); in test_simple_reshape()
64 gpu4.device(sycl_device)=gpu1.reshape(dim2).reshape(dim4); in test_simple_reshape()
105 TensorMap< Tensor<DataType, 3, DataLayout, IndexType> > gpu1(gpu_data1, dim1); in test_reshape_as_lvalue() local
111 gpu2.reshape(dim1).device(sycl_device)=gpu1; in test_reshape_as_lvalue()
114 gpu3.reshape(dim1).device(sycl_device)=gpu1; in test_reshape_as_lvalue()
153 TensorMap<Tensor<DataType, 5,DataLayout, IndexType>> gpu1(gpu_data1, tensorRange); in test_simple_slice() local
158 gpu2.device(sycl_device)=gpu1.slice(indices, sizes); in test_simple_slice()
169 gpu3.device(sycl_device)=gpu1.slice(indices2, sizes2); in test_simple_slice()
[all …]
|
| D | cxx11_tensor_math_sycl.cpp | 47 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu1(gpu_data1, tensorRange); in test_tanh_sycl() local
51 gpu2.device(sycl_device) = gpu1.tanh(); in test_tanh_sycl()
77 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu1(gpu_data1, tensorRange); in test_sigmoid_sycl() local
81 gpu2.device(sycl_device) = gpu1.sigmoid(); in test_sigmoid_sycl()
|
| D | cxx11_tensor_layout_swap_sycl.cpp | 43 TensorMap<Tensor<DataType, 3, ColMajor, IndexType>> gpu1(gpu_data1, tensorColRange); in test_simple_swap_sycl() local
47 gpu2.device(sycl_device)=gpu1.swap_layout(); in test_simple_swap_sycl()
86 TensorMap<Tensor<DataType, 3, ColMajor, IndexType>> gpu1(gpu_data1, tensorColRange); in test_swap_as_lvalue_sycl() local
90 gpu2.swap_layout().device(sycl_device)=gpu1; in test_swap_as_lvalue_sycl()
|
| D | cxx11_tensor_padding_sycl.cpp | 62 TensorMap<Tensor<DataType, 4,DataLayout,IndexType>> gpu1(gpu_data1, tensorRange); in test_simple_padding() local
70 gpu2.device(sycl_device)=gpu1.pad(paddings); in test_simple_padding()
116 TensorMap<Tensor<DataType, 4,DataLayout,IndexType>> gpu1(gpu_data1, tensorRange); in test_padded_expr() local
121 gpu2.device(sycl_device)=gpu1.pad(paddings).reshape(reshape_dims); in test_padded_expr()
|
| D | cxx11_tensor_shuffling_sycl.cpp | 49 TensorMap<Tensor<DataType, 4, DataLayout, IndexType>> gpu1(gpu_data1, in test_simple_shuffling_sycl() local
56 gpu2.device(sycl_device) = gpu1.shuffle(shuffles); in test_simple_shuffling_sycl()
86 gpu3.device(sycl_device) = gpu1.shuffle(shuffles); in test_simple_shuffling_sycl()
|
| D | cxx11_tensor_sycl.cpp | 46 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu1(gpu_data1, tensorRange); in test_sycl_mem_transfers() local
51 gpu1.device(sycl_device) = gpu1 * 3.14f; in test_sycl_mem_transfers()
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
|
| 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
71 gpu3.device(sycl_device)= gpu1.slice(indicesStart1, sizes) - gpu2.slice(indicesStart2, sizes); in test_image_op_sycl()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | device_util_test.cc | 72 const std::string gpu1 = "/job:worker/replica:1/task:2/device:GPU:1"; in TEST() local
77 devices.push_back(FakeDevice::Make(gpu1)); in TEST()
100 auto device_meta_2 = devices_attr.get(gpu1).dyn_cast<mlir::UnitAttr>(); in TEST()
187 const std::string gpu1 = "/job:worker/replica:0/task:0/device:GPU:1"; in TEST() local
205 DeviceNameUtils::ParseFullName(gpu1, &parsed_name); in TEST()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 112 // owned by that worker (e.g. cpu0, plus gpu0, gpu1, ..., gpu7). The
|