Searched refs:full_prec (Results 1 – 1 of 1) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_of_float16_cuda.cu | 46 Tensor<bool, 1> full_prec(num_elem); in test_cuda_numext() local
48 gpu_device.memcpyDeviceToHost(full_prec.data(), d_res_float, num_elem*sizeof(bool)); in test_cuda_numext()
53 VERIFY_IS_EQUAL(full_prec(i), half_prec(i)); in test_cuda_numext()
121 Tensor<float, 1> full_prec(num_elem); in test_cuda_unary() local
123 gpu_device.memcpyDeviceToHost(full_prec.data(), d_res_float, num_elem*sizeof(float)); in test_cuda_unary()
128 VERIFY_IS_APPROX(full_prec(i), half_prec(i)); in test_cuda_unary()
162 Tensor<float, 1> full_prec(num_elem); in test_cuda_elementwise() local
164 gpu_device.memcpyDeviceToHost(full_prec.data(), d_res_float, num_elem*sizeof(float)); in test_cuda_elementwise()
168 …std::cout << "Checking elemwise " << i << ": full prec = " << full_prec(i) << " vs half prec = " <… in test_cuda_elementwise()
169 … VERIFY_IS_APPROX(static_cast<Eigen::half>(full_prec(i)), static_cast<Eigen::half>(half_prec(i))); in test_cuda_elementwise()
[all …]
|