Searched defs:out_gpu (Results 1 – 3 of 3) sorted by relevance
43 TensorMap<Tensor<float, 0> > out_gpu(gpu_out_data); in test_full_reductions_sycl() local78 TensorMap<Tensor<float, 2> > out_gpu(gpu_out_data, reduced_tensorRange); in test_first_dim_reductions_sycl() local116 TensorMap<Tensor<float, 2> > out_gpu(gpu_out_data, reduced_tensorRange); in test_last_dim_reductions_sycl() local
95 TensorMap<Tensor<std::complex<float>, 0> > out_gpu(gpu_out_ptr); in test_cuda_sum_reductions() local132 TensorMap<Tensor<std::complex<float>, 0> > out_gpu(gpu_out_ptr); in test_cuda_product_reductions() local
44 TensorMap<Tensor<Type, 0, DataLayout> > out_gpu(gpu_out_ptr); in test_full_reductions() local