Home
last modified time | relevance | path

Searched refs:d_in1 (Results 1 – 4 of 4) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_complex_cuda.cu31 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()
Dcxx11_tensor_complex_cwise_ops_cuda.cu27 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()
Dcxx11_tensor_cuda.cu31 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 …]
Dcxx11_tensor_device.cu283 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()