Searched refs:test_cuda_simple_argmax (Results 1 – 1 of 1) sorted by relevance
24 void test_cuda_simple_argmax() in test_cuda_simple_argmax() function248 CALL_SUBTEST_1(test_cuda_simple_argmax<RowMajor>()); in test_cxx11_tensor_cuda()249 CALL_SUBTEST_1(test_cuda_simple_argmax<ColMajor>()); in test_cxx11_tensor_cuda()