Home
last modified time | relevance | path

Searched refs:gpu_in2_data (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_concatenation_sycl.cpp48 …DataType * gpu_in2_data = static_cast<DataType*>(sycl_device.allocate(right.dimensions().TotalSiz… in test_simple_concatenation() local
51 …Eigen::TensorMap<Eigen::Tensor<DataType, 3, DataLayout, IndexType>> gpu_in2(gpu_in2_data, rightRan… in test_simple_concatenation()
53 …sycl_device.memcpyHostToDevice(gpu_in2_data, right.data(),(right.dimensions().TotalSize())*sizeof(… in test_simple_concatenation()
113 sycl_device.deallocate(gpu_in2_data); in test_simple_concatenation()
140 …DataType * gpu_in2_data = static_cast<DataType*>(sycl_device.allocate(right.dimensions().TotalSiz… in test_concatenation_as_lvalue() local
145 …Eigen::TensorMap<Eigen::Tensor<DataType, 2, DataLayout, IndexType>> gpu_in2(gpu_in2_data, rightRan… in test_concatenation_as_lvalue()
149 …sycl_device.memcpyHostToDevice(gpu_in2_data, right.data(),(right.dimensions().TotalSize())*sizeof(… in test_concatenation_as_lvalue()
155 …sycl_device.memcpyDeviceToHost(right.data(), gpu_in2_data,(right.dimensions().TotalSize())*sizeof(… in test_concatenation_as_lvalue()
164 sycl_device.deallocate(gpu_in2_data); in test_concatenation_as_lvalue()
H A Dcxx11_tensor_forced_eval_sycl.cpp36 …DataType * gpu_in2_data = static_cast<DataType*>(sycl_device.allocate(in2.dimensions().TotalSize(… in test_forced_eval_sycl() local
44 …Eigen::TensorMap<Eigen::Tensor<DataType, 3, DataLayout, IndexType>> gpu_in2(gpu_in2_data, tensorRa… in test_forced_eval_sycl()
47 …sycl_device.memcpyHostToDevice(gpu_in2_data, in2.data(),(in2.dimensions().TotalSize())*sizeof(Data… in test_forced_eval_sycl()
61 sycl_device.deallocate(gpu_in2_data); in test_forced_eval_sycl()
H A Dcxx11_tensor_custom_op_sycl.cpp123 …DataType * gpu_in2_data = static_cast<DataType*>(sycl_device.allocate(in2.dimensions().TotalSize(… in test_custom_binary_op_sycl() local
128 TensorType gpu_in2(gpu_in2_data, tensorRange2); in test_custom_binary_op_sycl()
135 …sycl_device.memcpyHostToDevice(gpu_in2_data, in2.data(),(in2.dimensions().TotalSize())*sizeof(Data… in test_custom_binary_op_sycl()
153 sycl_device.deallocate(gpu_in2_data); in test_custom_binary_op_sycl()
H A Dcxx11_tensor_sycl.cpp195 …DataType * gpu_in2_data = static_cast<DataType*>(sycl_device.allocate(in2.size()*sizeof(DataType)… in test_sycl_computations() local
200 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu_in2(gpu_in2_data, tensorRange); in test_sycl_computations()
234 sycl_device.memcpyHostToDevice(gpu_in2_data, in2.data(),(in2.size())*sizeof(DataType)); in test_sycl_computations()
311 sycl_device.deallocate(gpu_in2_data); in test_sycl_computations()
/aosp_15_r20/external/eigen/unsupported/doc/examples/SYCL/
H A DCwiseMul.cpp38 …DataType * gpu_in2_data = static_cast<DataType*>(sycl_device.allocate(in2.size()*sizeof(DataType)… in main() local
43 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu_in2(gpu_in2_data, tensorRange); in main()
48 sycl_device.memcpyHostToDevice(gpu_in2_data, in2.data(),(in2.size())*sizeof(DataType)); in main()