Searched refs:t_result_bytes (Results 1 – 3 of 3) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_contract_cuda.cu | 44 std::size_t t_result_bytes = t_result.size() * sizeof(float); in test_cuda_contraction() local
52 cudaMalloc((void**)(&d_t_result), t_result_bytes); in test_cuda_contraction()
71 cudaMemcpy(t_result_gpu.data(), d_t_result, t_result_bytes, cudaMemcpyDeviceToHost); in test_cuda_contraction()
108 std::size_t t_result_bytes = sizeof(float); in test_scalar() local
116 cudaMalloc((void**)(&d_t_result), t_result_bytes); in test_scalar()
134 cudaMemcpy(t_result_gpu.data(), d_t_result, t_result_bytes, cudaMemcpyDeviceToHost); in test_scalar()
|
| D | cxx11_tensor_scan_cuda.cu | 36 std::size_t t_result_bytes = t_result.size() * sizeof(float); in test_cuda_cumsum() local
42 cudaMalloc((void**)(&d_t_result), t_result_bytes); in test_cuda_cumsum()
57 cudaMemcpy(t_result_gpu.data(), d_t_result, t_result_bytes, cudaMemcpyDeviceToHost); in test_cuda_cumsum()
|
| D | cxx11_tensor_cuda.cu | 271 std::size_t t_result_bytes = t_result.size() * sizeof(float); in test_cuda_contraction() local
279 cudaMalloc((void**)(&d_t_result), t_result_bytes); in test_cuda_contraction()
304 cudaMemcpy(t_result.data(), d_t_result, t_result_bytes, cudaMemcpyDeviceToHost); in test_cuda_contraction()
|