Searched refs:gpu_in_data (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_reduction_sycl.cpp | 39 …float* gpu_in_data = static_cast<float*>(sycl_device.allocate(in.dimensions().TotalSize()*sizeof(f… in test_full_reductions_sycl() local
42 TensorMap<Tensor<float, 2> > in_gpu(gpu_in_data, tensorRange); in test_full_reductions_sycl()
45 …sycl_device.memcpyHostToDevice(gpu_in_data, in.data(),(in.dimensions().TotalSize())*sizeof(float)); in test_full_reductions_sycl()
51 sycl_device.deallocate(gpu_in_data); in test_full_reductions_sycl()
74 …float* gpu_in_data = static_cast<float*>(sycl_device.allocate(in.dimensions().TotalSize()*sizeof(f… in test_first_dim_reductions_sycl() local
77 TensorMap<Tensor<float, 3> > in_gpu(gpu_in_data, tensorRange); in test_first_dim_reductions_sycl()
80 …sycl_device.memcpyHostToDevice(gpu_in_data, in.data(),(in.dimensions().TotalSize())*sizeof(float)); in test_first_dim_reductions_sycl()
89 sycl_device.deallocate(gpu_in_data); in test_first_dim_reductions_sycl()
112 …float* gpu_in_data = static_cast<float*>(sycl_device.allocate(in.dimensions().TotalSize()*sizeof(f… in test_last_dim_reductions_sycl() local
115 TensorMap<Tensor<float, 3> > in_gpu(gpu_in_data, tensorRange); in test_last_dim_reductions_sycl()
[all …]
|
| D | cxx11_tensor_broadcast_sycl.cpp | 47 …float * gpu_in_data = static_cast<float*>(sycl_device.allocate(input.dimensions().TotalSize()*siz… in test_broadcast_sycl() local
50 TensorMap<Tensor<float, 4>> gpu_in(gpu_in_data, in_range); in test_broadcast_sycl()
52 …sycl_device.memcpyHostToDevice(gpu_in_data, input.data(),(input.dimensions().TotalSize())*sizeof(f… in test_broadcast_sycl()
66 sycl_device.deallocate(gpu_in_data); in test_broadcast_sycl()
|