Home
last modified time | relevance | path

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

/external/eigen/unsupported/test/
Dcxx11_tensor_of_float16_cuda.cu142 float* d_float1 = (float*)gpu_device.allocate(num_elem * sizeof(float)); in test_cuda_elementwise() local
148 d_float1, num_elem); in test_cuda_elementwise()
172 gpu_device.deallocate(d_float1); in test_cuda_elementwise()
184 float* d_float1 = (float*)gpu_device.allocate(num_elem * sizeof(float)); in test_cuda_trancendental() local
194 Eigen::TensorMap<Eigen::Tensor<float, 1>, Eigen::Aligned> gpu_float1(d_float1, num_elem); in test_cuda_trancendental()
229 gpu_device.memcpyDeviceToHost(input1.data(), d_float1, num_elem*sizeof(float)); in test_cuda_trancendental()
255 gpu_device.deallocate(d_float1); in test_cuda_trancendental()
274 float* d_float1 = (float*)gpu_device.allocate(num_elem * sizeof(float)); in test_cuda_contractions() local
280 d_float1, rows, cols); in test_cuda_contractions()
311 gpu_device.deallocate(d_float1); in test_cuda_contractions()
[all …]