Home
last modified time | relevance | path

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

/external/eigen/unsupported/test/
Dcxx11_tensor_of_float16_cuda.cu143 float* d_float2 = (float*)gpu_device.allocate(num_elem * sizeof(float)); in test_cuda_elementwise() local
150 d_float2, num_elem); in test_cuda_elementwise()
173 gpu_device.deallocate(d_float2); in test_cuda_elementwise()
185 float* d_float2 = (float*)gpu_device.allocate(num_elem * sizeof(float)); in test_cuda_trancendental() local
195 Eigen::TensorMap<Eigen::Tensor<float, 1>, Eigen::Aligned> gpu_float2(d_float2, num_elem); in test_cuda_trancendental()
230 gpu_device.memcpyDeviceToHost(input2.data(), d_float2, num_elem*sizeof(float)); in test_cuda_trancendental()
256 gpu_device.deallocate(d_float2); in test_cuda_trancendental()
275 float* d_float2 = (float*)gpu_device.allocate(num_elem * sizeof(float)); in test_cuda_contractions() local
282 d_float2, rows, cols); in test_cuda_contractions()
312 gpu_device.deallocate(d_float2); in test_cuda_contractions()
[all …]