Searched refs:test_cuda_contraction (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_contract_cuda.cu | 27 void test_cuda_contraction(int m_size, int k_size, int n_size) in test_cuda_contraction() function
151 test_cuda_contraction<ColMajor>(k, 128, 128); in test_cuda_contraction_m()
152 test_cuda_contraction<RowMajor>(k, 128, 128); in test_cuda_contraction_m()
159 test_cuda_contraction<ColMajor>(128, k, 128); in test_cuda_contraction_k()
160 test_cuda_contraction<RowMajor>(128, k, 128); in test_cuda_contraction_k()
167 test_cuda_contraction<ColMajor>(128, 128, k); in test_cuda_contraction_n()
168 test_cuda_contraction<RowMajor>(128, 128, k); in test_cuda_contraction_n()
191 test_cuda_contraction<DataLayout>(m_sizes[i], n_sizes[j], k_sizes[k]); in test_cuda_contraction_sizes()
199 CALL_SUBTEST_1(test_cuda_contraction<ColMajor>(128, 128, 128)); in test_cxx11_tensor_cuda()
200 CALL_SUBTEST_1(test_cuda_contraction<RowMajor>(128, 128, 128)); in test_cxx11_tensor_cuda()
|
| D | cxx11_tensor_cuda.cu | 257 void test_cuda_contraction() in test_cuda_contraction() function
1222 CALL_SUBTEST_2(test_cuda_contraction<ColMajor>()); in test_cxx11_tensor_cuda()
1223 CALL_SUBTEST_2(test_cuda_contraction<RowMajor>()); in test_cxx11_tensor_cuda()
|