Home
last modified time | relevance | path

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

/external/eigen/unsupported/test/
Dcxx11_tensor_chipping_sycl.cpp47 DataType* gpu_data_tensor = static_cast<DataType*>(sycl_device.allocate(tensorBuffSize)); in test_static_chip_sycl() local
50 TensorMap<Tensor<DataType, 5, DataLayout,IndexType>> gpu_tensor(gpu_data_tensor, tensorRange); in test_static_chip_sycl()
53 sycl_device.memcpyHostToDevice(gpu_data_tensor, tensor.data(), tensorBuffSize); in test_static_chip_sycl()
169 sycl_device.deallocate(gpu_data_tensor); in test_static_chip_sycl()
196 DataType* gpu_data_tensor = static_cast<DataType*>(sycl_device.allocate(tensorBuffSize)); in test_dynamic_chip_sycl() local
199 TensorMap<Tensor<DataType, 5, DataLayout,IndexType>> gpu_tensor(gpu_data_tensor, tensorRange); in test_dynamic_chip_sycl()
202 sycl_device.memcpyHostToDevice(gpu_data_tensor, tensor.data(), tensorBuffSize); in test_dynamic_chip_sycl()
317 sycl_device.deallocate(gpu_data_tensor); in test_dynamic_chip_sycl()
346 DataType* gpu_data_tensor = static_cast<DataType*>(sycl_device.allocate(tensorBuffSize)); in test_chip_in_expr() local
350 TensorMap<Tensor<DataType, 5, DataLayout,IndexType>> gpu_tensor(gpu_data_tensor, tensorRange); in test_chip_in_expr()
[all …]
Dcxx11_tensor_inflation_sycl.cpp53 DataType* gpu_data_tensor = static_cast<DataType*>(sycl_device.allocate(tensorBuffSize)); in test_simple_inflation_sycl() local
56 TensorMap<Tensor<DataType, 4, DataLayout,IndexType>> gpu_tensor(gpu_data_tensor, tensorRange); in test_simple_inflation_sycl()
59 sycl_device.memcpyHostToDevice(gpu_data_tensor, tensor.data(), tensorBuffSize); in test_simple_inflation_sycl()
120 sycl_device.deallocate(gpu_data_tensor); in test_simple_inflation_sycl()
Dcxx11_tensor_patch_sycl.cpp55 DataType* gpu_data_tensor = static_cast<DataType*>(sycl_device.allocate(tensorBuffSize)); in test_simple_patch_sycl() local
58 TensorMap<Tensor<DataType, 4, DataLayout,IndexType>> gpu_tensor(gpu_data_tensor, tensorRange); in test_simple_patch_sycl()
61 sycl_device.memcpyHostToDevice(gpu_data_tensor, tensor.data(), tensorBuffSize); in test_simple_patch_sycl()
231 sycl_device.deallocate(gpu_data_tensor); in test_simple_patch_sycl()