Home
last modified time | relevance | path

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

/external/eigen/unsupported/test/
Dcxx11_tensor_complex_cwise_ops_cuda.cu28 std::complex<T>* d_in2; in test_cuda_complex_cwise_ops() local
31 cudaMalloc((void**)(&d_in2), complex_bytes); in test_cuda_complex_cwise_ops()
40 d_in2, kNumItems); in test_cuda_complex_cwise_ops()
88 cudaFree(d_in2); in test_cuda_complex_cwise_ops()
Dcxx11_tensor_complex_cuda.cu32 std::complex<float>* d_in2; in test_cuda_nullary() local
35 cudaMalloc((void**)(&d_in2), complex_bytes); in test_cuda_nullary()
38 cudaMemcpy(d_in2, in2.data(), complex_bytes, cudaMemcpyHostToDevice); in test_cuda_nullary()
46 d_in2, 2); in test_cuda_nullary()
69 cudaFree(d_in2); in test_cuda_nullary()
Dcxx11_tensor_cuda.cu32 float* d_in2; in test_cuda_nullary() local
34 cudaMalloc((void**)(&d_in2), tensor_bytes); in test_cuda_nullary()
36 cudaMemcpy(d_in2, in2.data(), tensor_bytes, cudaMemcpyHostToDevice); in test_cuda_nullary()
44 d_in2, 2); in test_cuda_nullary()
54 assert(cudaMemcpyAsync(new2.data(), d_in2, tensor_bytes, cudaMemcpyDeviceToHost, in test_cuda_nullary()
65 cudaFree(d_in2); in test_cuda_nullary()
80 float* d_in2; in test_cuda_elementwise_small() local
83 cudaMalloc((void**)(&d_in2), in2_bytes); in test_cuda_elementwise_small()
87 cudaMemcpy(d_in2, in2.data(), in2_bytes, cudaMemcpyHostToDevice); in test_cuda_elementwise_small()
95 d_in2, Eigen::array<Eigen::DenseIndex, 1>(2)); in test_cuda_elementwise_small()
[all …]
Dcxx11_tensor_device.cu284 float* d_in2; in test_gpu() local
287 cudaMalloc((void**)(&d_in2), in2_bytes); in test_gpu()
291 cudaMemcpy(d_in2, in2.data(), in2_bytes, cudaMemcpyHostToDevice); in test_gpu()
294 Eigen::TensorMap<Eigen::Tensor<float, 3> > gpu_in2(d_in2, 40,50,70); in test_gpu()