Searched refs:gpu_t_input (Results 1 – 2 of 2) sorted by relevance
49 gpu_t_input(d_t_input, Eigen::array<int, 3>(m_size, k_size, n_size)); in test_gpu_cumsum()53 gpu_t_result.device(gpu_device) = gpu_t_input.cumsum(1); in test_gpu_cumsum()
47 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu_t_input( in test_sycl_cumsum() local54 gpu_t_result.device(sycl_device) = gpu_t_input.cumsum(consume_dim, exclusive); in test_sycl_cumsum()