Searched defs:t_result_gpu (Results 1 – 2 of 2) sorted by relevance
36 Tensor<float, 2, DataLayout> t_result_gpu(m_size, n_size); in test_cuda_contraction() local100 Tensor<float, 0, DataLayout> t_result_gpu; in test_scalar() local
31 Tensor<float, 3, DataLayout> t_result_gpu(m_size, k_size, n_size); in test_cuda_cumsum() local