Searched refs:d_in1 (Results 1 – 4 of 4) sorted by relevance
/external/eigen/unsupported/test/ |
D | cxx11_tensor_complex_cuda.cu | 31 std::complex<float>* d_in1; in test_cuda_nullary() local 34 cudaMalloc((void**)(&d_in1), complex_bytes); in test_cuda_nullary() 37 cudaMemcpy(d_in1, in1.data(), complex_bytes, cudaMemcpyHostToDevice); in test_cuda_nullary() 44 d_in1, 2); in test_cuda_nullary() 56 assert(cudaMemcpyAsync(new1.data(), d_in1, complex_bytes, cudaMemcpyDeviceToHost, in test_cuda_nullary() 68 cudaFree(d_in1); in test_cuda_nullary()
|
D | cxx11_tensor_complex_cwise_ops_cuda.cu | 27 std::complex<T>* d_in1; in test_cuda_complex_cwise_ops() local 30 cudaMalloc((void**)(&d_in1), complex_bytes); in test_cuda_complex_cwise_ops() 38 d_in1, kNumItems); in test_cuda_complex_cwise_ops() 87 cudaFree(d_in1); in test_cuda_complex_cwise_ops()
|
D | cxx11_tensor_cuda.cu | 31 float* d_in1; in test_cuda_nullary() local 33 cudaMalloc((void**)(&d_in1), tensor_bytes); in test_cuda_nullary() 35 cudaMemcpy(d_in1, in1.data(), tensor_bytes, cudaMemcpyHostToDevice); in test_cuda_nullary() 42 d_in1, 2); in test_cuda_nullary() 52 assert(cudaMemcpyAsync(new1.data(), d_in1, tensor_bytes, cudaMemcpyDeviceToHost, in test_cuda_nullary() 64 cudaFree(d_in1); in test_cuda_nullary() 79 float* d_in1; in test_cuda_elementwise_small() local 82 cudaMalloc((void**)(&d_in1), in1_bytes); in test_cuda_elementwise_small() 86 cudaMemcpy(d_in1, in1.data(), in1_bytes, cudaMemcpyHostToDevice); in test_cuda_elementwise_small() 93 d_in1, Eigen::array<Eigen::DenseIndex, 1>(2)); in test_cuda_elementwise_small() [all …]
|
D | cxx11_tensor_device.cu | 283 float* d_in1; in test_gpu() local 286 cudaMalloc((void**)(&d_in1), in1_bytes); in test_gpu() 290 cudaMemcpy(d_in1, in1.data(), in1_bytes, cudaMemcpyHostToDevice); in test_gpu() 293 Eigen::TensorMap<Eigen::Tensor<float, 3> > gpu_in1(d_in1, 40,50,70); in test_gpu()
|