Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_mkldnn_pattern_matcher.py1611 def _qlinear_unary_cpu_test_helper( member in TestPatternMatcher
1653 self._qlinear_unary_cpu_test_helper((torch.randn((2, 4)),))
1663 self._qlinear_unary_cpu_test_helper(
1674 self._qlinear_unary_cpu_test_helper((torch.randn((2, 3, 4)),))
1684 self._qlinear_unary_cpu_test_helper(
1696 self._qlinear_unary_cpu_test_helper((torch.randn((2, 4)),), gelu)
1707 self._qlinear_unary_cpu_test_helper(