Searched refs:gpu_in3 (Results 1 – 2 of 2) sorted by relevance
51 TensorMap<Tensor<float, 3>> gpu_in3(gpu_in3_data, tensorRange); in test_sycl_cpu() local138 gpu_out.device(sycl_device) =(gpu_in1 > gpu_in1.constant(0.5f)).select(gpu_in2, gpu_in3); in test_sycl_cpu()
149 …Eigen::TensorMap<Eigen::Tensor<float, 3> > gpu_in3(d_in3, Eigen::array<Eigen::DenseIndex, 3>(72,53… in test_cuda_elementwise() local152 gpu_out.device(gpu_device) = gpu_in1 + gpu_in2 * gpu_in3; in test_cuda_elementwise()