Searched refs:t_result_gpu (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_contract_cuda.cu | 36 Tensor<float, 2, DataLayout> t_result_gpu(m_size, n_size); in test_cuda_contraction() local
71 cudaMemcpy(t_result_gpu.data(), d_t_result, t_result_bytes, cudaMemcpyDeviceToHost); in test_cuda_contraction()
73 if (fabs(t_result(i) - t_result_gpu(i)) < 1e-4f) { in test_cuda_contraction()
76 if (Eigen::internal::isApprox(t_result(i), t_result_gpu(i), 1e-4f)) { in test_cuda_contraction()
80 << " vs " << t_result_gpu(i) << std::endl; in test_cuda_contraction()
100 Tensor<float, 0, DataLayout> t_result_gpu; in test_scalar() local
134 cudaMemcpy(t_result_gpu.data(), d_t_result, t_result_bytes, cudaMemcpyDeviceToHost); in test_scalar()
135 if (fabs(t_result() - t_result_gpu()) > 1e-4f && in test_scalar()
136 !Eigen::internal::isApprox(t_result(), t_result_gpu(), 1e-4f)) { in test_scalar()
138 << " vs " << t_result_gpu() << std::endl; in test_scalar()
|
| D | cxx11_tensor_scan_cuda.cu | 31 Tensor<float, 3, DataLayout> t_result_gpu(m_size, k_size, n_size); in test_cuda_cumsum() local
57 cudaMemcpy(t_result_gpu.data(), d_t_result, t_result_bytes, cudaMemcpyDeviceToHost); in test_cuda_cumsum()
59 if (fabs(t_result(i) - t_result_gpu(i)) < 1e-4f) { in test_cuda_cumsum()
62 if (Eigen::internal::isApprox(t_result(i), t_result_gpu(i), 1e-4f)) { in test_cuda_cumsum()
66 << " vs " << t_result_gpu(i) << std::endl; in test_cuda_cumsum()
|