Home
last modified time | relevance | path

Searched refs:gpu_out_data (Results 1 – 4 of 4) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_reduction_sycl.cpp40 float* gpu_out_data =(float*)sycl_device.allocate(sizeof(float)); in test_full_reductions_sycl() local
43 TensorMap<Tensor<float, 0> > out_gpu(gpu_out_data); in test_full_reductions_sycl()
47 sycl_device.memcpyDeviceToHost(full_redux_gpu.data(), gpu_out_data, sizeof(float)); in test_full_reductions_sycl()
52 sycl_device.deallocate(gpu_out_data); in test_full_reductions_sycl()
75 …float* gpu_out_data = static_cast<float*>(sycl_device.allocate(redux_gpu.dimensions().TotalSize()*… in test_first_dim_reductions_sycl() local
78 TensorMap<Tensor<float, 2> > out_gpu(gpu_out_data, reduced_tensorRange); in test_first_dim_reductions_sycl()
82 …sycl_device.memcpyDeviceToHost(redux_gpu.data(), gpu_out_data, redux_gpu.dimensions().TotalSize()*… in test_first_dim_reductions_sycl()
90 sycl_device.deallocate(gpu_out_data); in test_first_dim_reductions_sycl()
113 …float* gpu_out_data = static_cast<float*>(sycl_device.allocate(redux_gpu.dimensions().TotalSize()*… in test_last_dim_reductions_sycl() local
116 TensorMap<Tensor<float, 2> > out_gpu(gpu_out_data, reduced_tensorRange); in test_last_dim_reductions_sycl()
[all …]
Dcxx11_tensor_sycl.cpp47 …float * gpu_out_data = static_cast<float*>(sycl_device.allocate(out.dimensions().TotalSize()*size… in test_sycl_cpu() local
52 TensorMap<Tensor<float, 3>> gpu_out(gpu_out_data, tensorRange); in test_sycl_cpu()
68 …sycl_device.memcpyDeviceToHost(out.data(), gpu_out_data ,(out.dimensions().TotalSize())*sizeof(flo… in test_sycl_cpu()
82 …sycl_device.memcpyDeviceToHost(out.data(), gpu_out_data,(out.dimensions().TotalSize())*sizeof(floa… in test_sycl_cpu()
96 …sycl_device.memcpyDeviceToHost(out.data(), gpu_out_data,(out.dimensions().TotalSize())*sizeof(floa… in test_sycl_cpu()
110 …sycl_device.memcpyDeviceToHost(out.data(), gpu_out_data,(out.dimensions().TotalSize())*sizeof(floa… in test_sycl_cpu()
124 …sycl_device.memcpyDeviceToHost(out.data(),gpu_out_data,(out.dimensions().TotalSize())*sizeof(float… in test_sycl_cpu()
139 …sycl_device.memcpyDeviceToHost(out.data(), gpu_out_data,(out.dimensions().TotalSize())*sizeof(floa… in test_sycl_cpu()
153 sycl_device.deallocate(gpu_out_data); in test_sycl_cpu()
Dcxx11_tensor_broadcast_sycl.cpp48 …float * gpu_out_data = static_cast<float*>(sycl_device.allocate(out.dimensions().TotalSize()*size… in test_broadcast_sycl() local
51 TensorMap<Tensor<float, 4>> gpu_out(gpu_out_data, out_range); in test_broadcast_sycl()
54 …sycl_device.memcpyDeviceToHost(out.data(), gpu_out_data,(out.dimensions().TotalSize())*sizeof(floa… in test_broadcast_sycl()
67 sycl_device.deallocate(gpu_out_data); in test_broadcast_sycl()
Dcxx11_tensor_forced_eval_sycl.cpp37 …float * gpu_out_data = static_cast<float*>(sycl_device.allocate(out.dimensions().TotalSize()*size… in test_forced_eval_sycl() local
45 Eigen::TensorMap<Eigen::Tensor<float, 3>> gpu_out(gpu_out_data, tensorRange); in test_forced_eval_sycl()
50 …sycl_device.memcpyDeviceToHost(out.data(), gpu_out_data,(out.dimensions().TotalSize())*sizeof(floa… in test_forced_eval_sycl()
62 sycl_device.deallocate(gpu_out_data); in test_forced_eval_sycl()