Searched refs:d_in3 (Results 1 – 1 of 1) sorted by relevance
133 float* d_in3; in test_cuda_elementwise() local137 cudaMalloc((void**)(&d_in3), in3_bytes); in test_cuda_elementwise()142 cudaMemcpy(d_in3, in3.data(), in3_bytes, cudaMemcpyHostToDevice); in test_cuda_elementwise()149 …Eigen::TensorMap<Eigen::Tensor<float, 3> > gpu_in3(d_in3, Eigen::array<Eigen::DenseIndex, 3>(72,53… in test_cuda_elementwise()167 cudaFree(d_in3); in test_cuda_elementwise()