Searched refs:d_float (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_cast_float16_cuda.cu | 32 float* d_float = (float*)gpu_device.allocate(num_elem * sizeof(float)); in test_cuda_conversion() local
37 d_float, num_elem); in test_cuda_conversion()
43 gpu_device.memcpyHostToDevice(d_float, floats.data(), num_elem*sizeof(float)); in test_cuda_conversion()
50 gpu_device.memcpyDeviceToHost(initial.data(), d_float, num_elem*sizeof(float)); in test_cuda_conversion()
58 gpu_device.deallocate(d_float); in test_cuda_conversion()
|
| D | cxx11_tensor_of_float16_cuda.cu | 30 float* d_float = (float*)gpu_device.allocate(num_elem * sizeof(float)); in test_cuda_numext() local
35 d_float, num_elem); in test_cuda_numext()
56 gpu_device.deallocate(d_float); in test_cuda_numext()
70 float* d_float = (float*)gpu_device.allocate(num_elem * sizeof(float)); in test_cuda_conversion() local
75 d_float, num_elem); in test_cuda_conversion()
87 gpu_device.memcpyDeviceToHost(initial.data(), d_float, num_elem*sizeof(float)); in test_cuda_conversion()
94 gpu_device.deallocate(d_float); in test_cuda_conversion()
105 float* d_float = (float*)gpu_device.allocate(num_elem * sizeof(float)); in test_cuda_unary() local
110 d_float, num_elem); in test_cuda_unary()
131 gpu_device.deallocate(d_float); in test_cuda_unary()
[all …]
|