Searched refs:make_cpp_wrapper_test (Results 1 – 1 of 1) sorted by relevance
47 def make_cpp_wrapper_test(orig_test, **extra_args): function138 test_softmax_cpp_wrapper = make_cpp_wrapper_test(290 test_reduction_scratch_buffer_cpp_wrapper = make_cpp_wrapper_test(297 make_cpp_wrapper_test(test_reduction_scratch_buffer, force_multi_kernel=2)301 make_cpp_wrapper_test(test_reduction_scratch_buffer, force_multi_kernel=3)