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 …]
|