Searched defs:gpu_in_data (Results 1 - 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_broadcast_sycl.cpp | 47 float * gpu_in_data = static_cast<float*>(sycl_device.allocate(input.dimensions().TotalSize()*sizeof(float))); local
50 TensorMap<Tensor<float, 4>> gpu_in(gpu_in_data, in_range);
52 sycl_device.memcpyHostToDevice(gpu_in_data, input.data(),(input.dimensions().TotalSize())*sizeof(float));
66 sycl_device.deallocate(gpu_in_data);
|
| H A D | cxx11_tensor_reduction_sycl.cpp | 39 float* gpu_in_data = static_cast<float*>(sycl_device.allocate(in.dimensions().TotalSize()*sizeof(float))); local
42 TensorMap<Tensor<float, 2> > in_gpu(gpu_in_data, tensorRange);
45 sycl_device.memcpyHostToDevice(gpu_in_data, in.data(),(in.dimensions().TotalSize())*sizeof(float));
51 sycl_device.deallocate(gpu_in_data);
74 float* gpu_in_data = static_cast<float*>(sycl_device.allocate(in.dimensions().TotalSize()*sizeof(float))); local
77 TensorMap<Tensor<float, 3> > in_gpu(gpu_in_data, tensorRange);
80 sycl_device.memcpyHostToDevice(gpu_in_data, in.data(),(in.dimensions().TotalSize())*sizeof(float));
89 sycl_device.deallocate(gpu_in_data);
112 float* gpu_in_data = static_cast<float*>(sycl_device.allocate(in.dimensions().TotalSize()*sizeof(float))); local
115 TensorMap<Tensor<float, 3> > in_gpu(gpu_in_data, tensorRang
[all...] |
Completed in 94 milliseconds