Home
last modified time | relevance | path

Searched refs:test_gpu_contraction (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_contract_gpu.cu26 void test_gpu_contraction(int m_size, int k_size, int n_size) in test_gpu_contraction() function
150 test_gpu_contraction<ColMajor>(k, 128, 128); in test_gpu_contraction_m()
151 test_gpu_contraction<RowMajor>(k, 128, 128); in test_gpu_contraction_m()
158 test_gpu_contraction<ColMajor>(128, k, 128); in test_gpu_contraction_k()
159 test_gpu_contraction<RowMajor>(128, k, 128); in test_gpu_contraction_k()
166 test_gpu_contraction<ColMajor>(128, 128, k); in test_gpu_contraction_n()
167 test_gpu_contraction<RowMajor>(128, 128, k); in test_gpu_contraction_n()
190 test_gpu_contraction<DataLayout>(m_sizes[i], n_sizes[j], k_sizes[k]); in test_gpu_contraction_sizes()
198 CALL_SUBTEST_1(test_gpu_contraction<ColMajor>(128, 128, 128)); in EIGEN_DECLARE_TEST()
199 CALL_SUBTEST_1(test_gpu_contraction<RowMajor>(128, 128, 128)); in EIGEN_DECLARE_TEST()
H A Dcxx11_tensor_gpu.cu258 void test_gpu_contraction() in test_gpu_contraction() function
1548 CALL_SUBTEST_2(test_gpu_contraction<ColMajor>()); in EIGEN_DECLARE_TEST()
1549 CALL_SUBTEST_2(test_gpu_contraction<RowMajor>()); in EIGEN_DECLARE_TEST()